У меня возникли проблемы с этим вопросом.
Когда я пытаюсь построить график рассеяния с помощью ListedColormap, следующий код выглядит нормально.
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from matplotlib.colors import ListedColormap
X_set = np.arange(20).reshape(10, 2)
y_set = np.array([1,1,0,1,1,0,1,1,0,1])
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(('blue', 'green'))(i), label = j)
Figure1
Однако, когда я изменил массив y_set на
y_set = np.array([0,1,1,1,0,0,1,1,0,1])
Он строился так: Figure2 .
Почему-то кажется, что ListedColormap не делает 'больше не работает.
Почему это происходит?