【erro】ValueError: c of shape (1, 300) not acceptable as a color sequence for x with size 300, y with

吴恩达深度学习作业 第二个课程第一周

init_utils.plot_decision_boundary(lambda x: init_utils.predict_dec(parameters, x.T), train_X, train_Y)

1.报错：

ValueError: c of shape (1, 300) not acceptable as a color sequence for x with size 300, y with

2.改正：

init_utils.plot_decision_boundary(lambda x: init_utils.predict_dec(parameters, x.T), train_X, np.squeeze(train_Y))

3.报错原因

这是因为matplotlib.pyplot接收的color sequence 参数c应该是单个值，直接传Y或者Y.train传的是shape(1, size)的矩阵。所以要使用np.squeeze(train_Y)这个函数的作用是去掉矩阵里维度为1的维度。

例：(1, 300)的矩阵经由np.squeeze处理后变成300; (200, 1, 300)的矩阵经由np.squeeze处理后变成(200, 300)。

numpy.squeeze
numpy.squeeze(a, axis=None)[source]
Remove single-dimensional entries from the shape of an array.

Parameters:	
a : array_like

Input data.

axis : None or int or tuple of ints, optional

New in version 1.7.0.

Selects a subset of the single-dimensional entries in the shape. If an axis is selected with shape entry greater than one, an error is raised.

Returns:	
squeezed : ndarray

The input array, but with all or a subset of the dimensions of length 1 removed. This is always a itself or a view into a.

>>>
>>> x = np.array([[[0], [1], [2]]])
>>> x.shape
(1, 3, 1)
>>> np.squeeze(x).shape
(3,)
>>> np.squeeze(x, axis=(2,)).shape
(1, 3)