pytorch利用scatter画散点图报错：ValueError:cofshape(200,1)notacceptableasacolorsequenceforx..

pytorch利用 scatter画散点图报错 &＃xff1a;ValueError: c of shape (200, 1) not acceptable as a color sequence for x with size 200, y with size 200

报错的全部内容为&＃xff1f;File "D:\ProgramData\Anaconda3\lib\site-packages\matplotlib\colors.py", line 223, in _to_rgba_no_colorcycleraise ValueError("RGBA sequence should have length 3 or 4")
ValueError: RGBA sequence should have length 3 or 4During handling of the above exception, another exception occurred:Traceback (most recent call last):File "E:/python/pyTorch/Learn_0.4_Pytorch/classify/classify.py", line 22, in plt.scatter(x_data.numpy()[:,0], x_data.numpy()[:,1], c&＃61;y_data.numpy(), s&＃61;200, lw&＃61;0, cmap&＃61;&＃39;RdYlGn&＃39;)File "D:\ProgramData\Anaconda3\lib\site-packages\matplotlib\pyplot.py", line 3470, in scatteredgecolors&＃61;edgecolors, data&＃61;data, **kwargs)File "D:\ProgramData\Anaconda3\lib\site-packages\matplotlib\__init__.py", line 1855, in innerreturn func(ax, *args, **kwargs)File "D:\ProgramData\Anaconda3\lib\site-packages\matplotlib\axes\_axes.py", line 4279, in scatter.format(c.shape, x.size, y.size))
ValueError: c of shape (200, 1) not acceptable as a color sequence for x with size 200, y with size 200Process finished with exit code 1

解决办法&＃xff1a;

将代码&＃xff1a;plt.scatter(x_data.numpy()[:,0], x_data.numpy()[:,1], c&＃61;y_data.numpy(), s&＃61;200, lw&＃61;0, cmap&＃61;&＃39;RdYlGn&＃39;)

修改为&＃xff1a;plt.scatter(x_data.numpy()[:,0], x_data.numpy()[:,1],c&＃61;np.squeeze(y_data.numpy()), s&＃61;200, lw&＃61;0, cmap&＃61;&＃39;RdYlGn&＃39;)Note&＃xff1a; 别忘记导入 numpy ,即需要在代码开头有这样一句代码&＃xff1a;import numpy as np