python - 只能将 '.sparse' 访问器与稀疏数据一起使用

Question

我使用稀疏 = True 参数对数据执行了 OHE - 这似乎没有做任何事情？

我尝试过:

# One Hot Encoding
df_ohe = pd.get_dummies(df, columns=cats, drop_first=True, sparse=True)
df_ohe = df_ohe.sparse.to_coo().tocsr() #Explicitely convert
df_ohe.memory_usage().sum()

...返回

AttributeError: Can only use the '.sparse' accessor with Sparse data.

如果有帮助，我们将不胜感激。谢谢!

Answer 1

您将需要使用导入 csr_matrix 方法(将 numpy 数组转换为稀疏矩阵)

from scipy.sparse import csr_matrix

然后你就可以写

df_ohe = pd.get_dummies(df, columns=cats, drop_first=True)
df_ohe = csr_matrix(df_ohe.values)

因此请注意，这里我从 get_dummies 方法中删除了 sparse=True，然后更改了转换为稀疏矩阵的语法。