python - 使用 plt.contour : Input z must be a 2D array 时出错

Question

我在使用 matplotlib 的 plt.contour 函数时收到错误消息“Input z must be a 2D array”。

我已经尝试过使用 meshgrid 但它不起作用，我无法真正找到问题所在。 E = np.zeros(2500) 是一个数组，所以我找不到问题所在。

import numpy as np
import numpy.random as rd
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from mkgrid2d import *

W1 = np.linspace(-1.0,3.0,num = 50)
W0 = np.linspace(-2.0,4.0,num = 50)

w11 = 1.00
w00=1.0

w = np.array([[w11],[w00]],dtype=float)

mew = 0
sigma = np.sqrt(2)


Npts=50;rd.seed(1)
x1=rd.rand(1,Npts)*10.-5.0 #Npts uniformemente distribuídos
r = np.random.normal(mew, sigma, 50)*2.-1.0#ruído gaussiano distribuído

X = np.vstack((x1,np.ones((1,x1.shape[1]))))
X = X.astype('float') #converter para float

N=Npts 

y = np.dot(w.T,X) + r
E= np.zeros(2500) 

“E”将是轮廓的 Z

count = 0
for i in range (len(W1)):
    for j in range (len (W0)):
        w1 = np.array([W1[i],W0[j]])
        yn = np.dot(w1.T,X)
        E[count] = (1./50)*(np.sum((y-yn)**2))
        count +=1

plt.figure()
CS = plt.contour(W1,W0,E)
plt.clabel(CS, inline=1, fontsize=10)
plt.title('Simplest default with labels')
        

Answer 1

如错误消息所述，您需要E 数组是2D，形状为(50,50)，而不是1D形状数组 (2500)。

有几种解决方法:

1. 赋值后 reshape 你的E

   E = E.reshape(len(W1),len(W0))
   

2. 首先创建具有正确形状的 E，然后使用您的 i 和 j 对其进行索引

   E = np.zeros((len(W1),len(W0)))
   
   for i in range (len(W1)):
       for j in range (len (W0)):
           w1 = np.array([W1[i],W0[j]])
           yn = np.dot(w1.T,X)
           E[i][j] = (1./50)*(np.sum((y-yn)**2))