Ошибка получения при построении графика в python

Question

, когда я пытался сделать перечисление точек данных моих наборов данных и нанести их на график таким образом, чтобы цвет мог его дифференцировать, это вызывает у меня ошибку

'c' выглядит аргументом как одна цифра c последовательность RGB или RGBA, чего следует избегать, поскольку отображение значений будет иметь приоритет в случае, если его длина совпадает с 'x' & 'y'. Пожалуйста, используйте двумерный массив с одной строкой, если вы действительно хотите задать одинаковое значение RGB или RGBA для всех точек.

вот мой код, который использовался для построения моего графика

from matplotlib.colors import ListedColormap

X_set, y_set = x_test, y_test

X1, X2 = np.meshgrid(np.arange(start = X_set[:, 0].min()-1 , stop = X_set[:, 0].max()+1 , step = 0.01),
                     np.arange(start = X_set[:, 1].min()-1 , stop = X_set[:, 1].max()+1 , step = 0.01))

plt.contourf(X1, X2, classifier.predict(np.array([X1.ravel(), X2.ravel()]).T).reshape(X1.shape),
             alpha = 0.65, cmap = ListedColormap(('red', 'green')))

plt.xlim(X1.min(), X1.max())

plt.ylim(X2.min(), X2.max())

for i, j in enumerate(np.unique(y_set)):

    plt.scatter(X_set[y_set == j, 0], X_set[y_set == j, 1],

                c = ListedColormap(('red', 'green'))(i), label = j)

plt.title('Logistic Regression (Training set)')

plt.xlabel('Age')

plt.ylabel('Estimated Salary')

plt.legend()

plt.show()