如何在python中绘制二维数组？

Question

我有

的数据集

data = [[1,2,3], [4,5,6], [7,8,9]].

并致电

plot(data)
plot.show()

然后将y轴视为内部数组的值。

我想要的是

f(0,0) = 1, f(0,1) = 2, f(1,2) = 3, 
f(1,0) = 4, f(1,1) = 5, f(1,2) = 6,
f(2,0) = 7, f(2,1) = 8, f(2,2) = 9

问题是，如何将y轴更改为数组的索引而不是数组的值？

Answer 1

一些事情：

1）Python没有2D，f[i,j]索引表示法，但为了得到它你可以使用numpy。从您的示例中选择任意索引对：

import numpy as np
f = np.array(data)
print f[1,2]
# 6
print data[1][2] 
# 6

2）然后你可以做的情节：

plt.imshow(f, interpolation="nearest", origin="upper")
plt.colorbar()
plt.show()

所以在这里你有代表性的颜色，你的f数组中有数字。

我在这里指定了origin="upper"。通常人们想要数据数组底部的(0,0)点（而不是图像），但是你在左上角用(0,0)写出你的数组，这就是"upper"确实。这也是默认的，顺便说一句，但明确使用可能会明确表明有一个选项。

Answer 2

您使用什么模块进行绘图？ Matplotlib？

你想要的是2d直方图。 有各种实现。我建议那个带有numpy的那个 http://docs.scipy.org/doc/numpy/reference/generated/numpy.histogram2d.html