如何从 anxn 矩阵生成等值线图？

Question

我有一个名为 o_pottial 的 xn 矩阵，其中填充了一些值，并希望从中生成等值线图。我尝试了几种方法但没有成功。这就是我现在所拥有的。

n = 20
x = arange(0,n-1)
y = arange(0,n-1)
plt.contourf(x, y, o_potential[x][y])
plt.show()

o_pottial 看起来像这样

o_potential = [[ -1.,          -1.,          -1.,          -1.,          -1.        ],
 [ 10.,           4.12244898,   2.7755102,    4.12244898,  10.        ],
 [ 10.,           5.7755102,    4.24489796,   5.7755102,   10.        ],
 [ 10.,           4.12244898,   2.7755102,    4.12244898,  10.        ],
 [ -1.,          -1.,          -1.,          -1.,          -1.        ]]

我收到以下错误消息：“用作索引的数组必须是整数（或布尔）类型。”

从昨天开始我就一直在为此苦苦挣扎。我尝试了几乎所有的Google结果，但无法解决问题。感谢您的帮助！

Answer 1

尝试：

n = 20
x = arange(n)
y = arange(n)
X, Y = meshgrid(x, y)
plt.contourf(X, Y, o_potential)
plt.show()

或者只是

plt.contourf(o_potential)
plt.show()

很难确切地说问题是什么，因为您没有显示 o_pottial，但上面的方法可能会起作用。

Answer 2

你也可以尝试这个：

import matplotlib.pyplot as plt
plt.imshow(o_potential, cmap='viridis')