将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代码时，这看起来像是一个错误。 当您尝试将dok_matrix中的某个行和列设置为零时，它会出现错误，因为它尝试删除该行和列的值而不检查它是否存在。

在回答你的问题时，你正在以不雅的方式做的事情正是__setitem__方法目前用你优雅的方法做的事情（在几次isinstance检查之后，什么不是）。 如果你想使用优雅的方式，你可以通过在Lib/site-packages/scipy/sparse/更改Lib/site-packages/scipy/sparse/来修复我在你自己的Scipy包中提到的错误。

if value==0:

至

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

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

Answer 2

我认为这个bug已经在Scipy 0.8.0中得到修复