4

我有一个格式的 numpy 向量列表：

    [array([[-0.36314615,  0.80562619, -0.82777381, ...,  2.00876354,2.08571887, -1.24526026]]), 
     array([[ 0.9766923 , -0.05725135, -0.38505339, ...,  0.12187988,-0.83129255,  0.32003683]]),
     array([[-0.59539878,  2.27166874,  0.39192573, ..., -0.73741573,1.49082653,  1.42466276]])]

此处仅显示列表中的 3 个向量。我有100个。。

一个向量中的最大元素数约为 1000 万

列表中的所有数组都具有不相等的元素数量，但最大元素数量是固定的。是否可以在 python 中使用这些向量创建一个稀疏矩阵，以便我用零代替小于最大大小的向量的元素？

4

3 回答 3

3

尝试这个：

from scipy import sparse
M = sparse.lil_matrix((num_of_vectors, max_vector_size))

for i,v in enumerate(vectors):
     M[i, :v.size] = v

然后看看这个页面：http ://docs.scipy.org/doc/scipy/reference/sparse.html

该lil_matrix格式适用于构建矩阵，但您需要将其转换为不同的格式，就像csr_matrix在对它们进行操作之前一样。

于 2013-08-26T21:35:16.890 回答
2

在这种方法中，您将低于阈值的元素替换为0，然后从中创建一个稀疏矩阵。我建议这样做，coo_matrix因为根据您的目的转换为其他类型是最快的。然后你可以scipy.sparse.vstack()让他们建立你的矩阵来计算列表中的所有元素：

import scipy.sparse as ss
import numpy as np

old_list = [np.random.random(100000) for i in range(5)]

threshold = 0.01
for a in old_list:
    a[np.absolute(a) < threshold] = 0
old_list = [ss.coo_matrix(a) for a in old_list]
m = ss.vstack( old_list )
于 2013-08-26T21:37:09.730 回答
1

有点复杂，但我可能会这样做：

>>> import scipy.sparse as sps
>>> a = [np.arange(5), np.arange(7), np.arange(3)]
>>> lens = [len(j) for j in a]
>>> cols = np.concatenate([np.arange(j) for j in lens])
>>> rows = np.concatenate([np.repeat(j, len_) for j, len_ in enumerate(lens)])
>>> data = np.concatenate(a)
>>> b = sps.coo_matrix((data,(rows, cols)))
>>> b.toarray()
array([[0, 1, 2, 3, 4, 0, 0],
       [0, 1, 2, 3, 4, 5, 6],
       [0, 1, 2, 0, 0, 0, 0]])
于 2013-08-26T21:44:28.210 回答