DeepLearning: ValueError: RGBA sequence should have length 3 or 4

利用plot.scatter绘制 散点图(Scatter) 时，出现以下错误：

Traceback (most recent call last):
  File "D:\Anaconda\envs\gpu-env\lib\site-packages\matplotlib\axes\_axes.py", line 4284, in _parse_scatter_color_args
    colors = mcolors.to_rgba_array(c)
  File "D:\Anaconda\envs\gpu-env\lib\site-packages\matplotlib\colors.py", line 294, in to_rgba_array
    result[i] = to_rgba(cc, alpha)
  File "D:\Anaconda\envs\gpu-env\lib\site-packages\matplotlib\colors.py", line 177, in to_rgba
    rgba = _to_rgba_no_colorcycle(c, alpha)
  File "D:\Anaconda\envs\gpu-env\lib\site-packages\matplotlib\colors.py", line 244, in _to_rgba_no_colorcycle
    raise ValueError("RGBA sequence should have length 3 or 4")
ValueError: RGBA sequence should have length 3 or 4

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "G:/pycharm Project/PytorchLearning/neural_network/week3/Double_NN.py", line 17, in <module>
    plot.scatter(X[0, :], X[1, :], c=Y, s=40, cmap=plot.cm.Spectral)
  File "D:\Anaconda\envs\gpu-env\lib\site-packages\matplotlib\pyplot.py", line 2841, in scatter
    None else {}), **kwargs)
  File "D:\Anaconda\envs\gpu-env\lib\site-packages\matplotlib\__init__.py", line 1599, in inner
    return func(ax, *map(sanitize_sequence, args), **kwargs)
  File "D:\Anaconda\envs\gpu-env\lib\site-packages\matplotlib\axes\_axes.py", line 4453, in scatter
    get_next_color_func=self._get_patches_for_fill.get_next_color)
  File "D:\Anaconda\envs\gpu-env\lib\site-packages\matplotlib\axes\_axes.py", line 4297, in _parse_scatter_color_args
    .format(nc=n_elem, xs=xsize, ys=ysize)
ValueError: 'c' argument has 1 elements, which is not acceptable for use with 'x' with size 400, 'y' with size 400.

原程序如下：

X, Y = load_planar_dataset()
plot.scatter(X[0, :], X[1, :], c=Y, s=40, cmap=plot.cm.Spectral)

plot.show()

修改c=np.squeeze(Y)

X, Y = load_planar_dataset()
plot.scatter(X[0, :], X[1, :], c=np.squeeze(Y), s=40, cmap=plot.cm.Spectral)

plot.show()

绘图成功！