将 numpy 数组添加到 scipy.sparse.dok_matrix

Question

我有一个 scipy.sparse.dok_matrix （尺寸 mxn），想要添加一个长度为 m 的平面 numpy 数组。

for col in xrange(n):
    dense_array = ...
    dok_matrix[:,col] = dense_array

但是，当此代码尝试删除不存在的键 (del self[(i,j)]) 时，它会在 dok_matrix.__setitem__ 中引发异常。

所以，现在我正在以不优雅的方式这样做：

for col in xrange(n):
    dense_array = ...
    for row in dense_array.nonzero():
        dok_matrix[row, col] = dense_array[row]

这感觉非常低效。 那么，最有效的方法是什么？

谢谢！

Answer 1

我认为这个错误已在 Scipy 0.8.0 中修复

Answer 2

我很惊讶你的不优雅的方式没有与切片方式相同的问题。在查看 Scipy 代码后，这对我来说似乎是一个错误。当您尝试将 dok_matrix 中的某个行和列设置为零（当它已经为零时）时，会出现错误，因为它尝试删除该行和列的值而不检查它是否存在。

在回答您的问题时，您以不优雅的方式所做的正是 __setitem__ 方法当前使用优雅方法所做的事情（经过几次 isinstance 检查以及其他检查之后）。如果你想使用优雅的方式，你可以通过打开 Lib/site-packages/scipy/sparse/ 中的 dok.py 并将第 222 行更改为

if value==0:

然后

if value==0 and self.has_key((i,j)):

你可以使用优雅的方式，它应该工作得很好。我去提交了一个错误修复，但这已经在下一个版本中修复了，这就是修复的方式。