是否可以在python中堆叠稀疏和密集的numpy数组?我知道这可以使用vstack / hstack为密集的numpy数组完成。我有一些列,我想添加到稀疏矩阵,以增加特征向量的数量
答案 0 :(得分:10)
是的,您可以使用scipy.sparse.vstack
和scipy.sparse.hstack
,方法与使用numpy.vstack
和numpy.hstack
密集阵列的方式相同。
示例:
from scipy.sparse import coo_matrix
m = coo_matrix(np.array([[0,0,1],[1,0,0],[1,0,0]]))
a = np.ones(m.shape)
使用np.vstack
:
np.vstack((a,m))
#ValueError: all the input array dimensions except for the concatenation axis must match exactly
使用scipy.sparse.vstack
:
scipy.sparse.vstack((a,m))
#<6x3 sparse matrix of type '<type 'numpy.float64'>'
# with 12 stored elements in COOrdinate format>