Numexpr 无法识别浮点类型（稀疏矩阵）答案

【问题标题】：Numexpr doesn't recognize float type (sparse matrix)Numexpr 无法识别浮点类型（稀疏矩阵）
【发布时间】：2015-11-20 10:41:18
【问题描述】：

我想评估 python (2.7) 中 numexpr 模块的性能。为此，我创建了一个大小为 (10^5, 10^5) 的随机稀疏矩阵。但是，下面的脚本已经在表达式求值步骤中抛出了一个错误，说它不能识别对象类型。

我做错了什么？

代码：

import timeit
import scipy.sparse as sps
import numpy as np
import numexpr as ne

test_matrix = sps.rand(1e4, 1e4, density=0.01, format='coo', dtype = np.float32)
ne.evaluate('sum(test_matrix, axis = 1)')

setup = 'import numexpr as ne; import numpy as np'
print min(timeit.Timer('ne.evaluate(sum(test_matrix, axis = 1))', setup=setup).repeat(7, 1000))

错误：

Traceback（最近一次调用最后一次）：

File "benchmark_expressmath.py", line 19, in <module>
ne.evaluate('sum(test_matrix, axis = 1)')
File "C:\Users\blahblah\AppData\Local\Continuum\Anaconda\lib\site-packages\numexpr\necompiler.py", line 756, in evaluate
signature = [(name, getType(arg)) for (name, arg) in zip(names, arguments)]
File "C:\Users\blahblah\AppData\Local\Continuum\Anaconda\lib\site-packages\numexpr\necompiler.py", line 654, in getType
raise ValueError("unknown type %s" % a.dtype.name)
ValueError: unknown type object

【问题讨论】：

标签： python python-2.7 numpy timeit numexpr

【解决方案1】：

numexpr 期望变量是 numpy 数组。它不处理 scipy 的稀疏矩阵。（例如，请参阅此电子邮件主题：http://numpy-discussion.10968.n7.nabble.com/ANN-numexpr-2-3-final-released-td36154.html）

【讨论】：

你知道任何类似的 Python 包可以处理稀疏矩阵吗？
有pysparse，但是没试过，不知道性能和scipy比怎么样。
pysparse 不具备 AFAIK 类似 numexpr 的功能。
@pv.，不，我不这么认为，但如果 OP 调查 numexpr 的原因是为了提高性能，并且如果结果证明 pysparse 比 scipy 快得多，则需要numexpr 会减少。