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

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

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

o_potential看起来像这样

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结果，但无法解决问题。感谢您的帮助！

Try:

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

or just

plt.contourf(o_potential)
plt.show()

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