Convert Pandas dataframe to Sparse Numpy Matrix directly

后端 未结 1 950
太阳男子
太阳男子 2020-12-04 16:28

I am creating a matrix from a Pandas dataframe as follows:

dense_matrix = np.array(df.as_matrix(columns = None), dtype=bool).astype(np.int)

相关标签:
1条回答
  • 2020-12-04 17:22

    df.values is a numpy array, and accessing values that way is always faster than np.array.

    scipy.sparse.csr_matrix(df.values)
    

    You might need to take the transpose first, like df.values.T. In DataFrames, the columns are axis 0.

    0 讨论(0)
提交回复
热议问题