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

作为未来谷歌员工的注意事项，还有pcolormesh 和pcolorfast。

pcolormesh 的文档指出：

pcolormesh 类似于 pcolor()，但使用不同的机制和 返回一个不同的对象； pcolor 返回一个 PolyCollection 但 pcolormesh 返回一个 QuadMesh。它快得多，所以它几乎 总是首选大型数组。

imshow 应该更快，但在例如非直线轴。

请参阅 this page，了解 pcolor、pcolormesh 和 imshow 之间的良好比较。

Answer 2

imshow 会快得多。 pcolor 返回一个 PolyCollection，如果有一百万个元素，它会相当慢，而 imshow 只是一个图像。

请注意，pcolor 中的索引与 imshow 略有不同，但您可能无需担心，具体取决于您使用 pcolor 的方式。此外，当从 pcolor 转到 imshow 时，通常希望在 imshow 中设置 interpolation="nearest"（但对于如此大的图像，这也可能无关紧要）。