python - matplotlib.pcolor 非常慢。备择方案？

Question

我想绘制一个二维数组(大约 1000x1000)，其值对应于色标。所以我使用了 matplotlib.pcolor，它就是这样做的，但出于某种原因，当它达到这些尺寸时它非常慢(比如 2 分钟左右只是为了绘图)。这是什么原因？将浮点值转换为 int16 左右会有帮助吗？ pcolor 有其他替代品吗？

from pylab import *

data=genfromtxt('data.txt',autostrip=True, case_sensitive=True)
pcolor(data,cmap='hot')
colorbar()
show()

data.txt 包含数组。加载过程确实需要几秒钟，但主要的计算时间肯定由 pcolor() 和 show() 函数使用(每个大约 60-90 秒)。

Answer 1

作为给 future google 员工的注意事项，还有 pcolormesh和 pcolorfast .

pcolormesh 的文档指出:

pcolormesh is similar to pcolor(), but uses a different mechanism and returns a different object; pcolor returns a PolyCollection but pcolormesh returns a QuadMesh. It is much faster, so it is almost always preferred for large arrays.

imshow应该更快，但在例如方面不太灵活非直线轴。

参见 this page pcolor、pcolormesh 和 imshow 之间的比较。