使用matplotlib.pyploy为不同的组绘制具有不同颜色的散点图

Question

我正在尝试使用matplotlib.pyplot（作为plt导入）并使用以下命令为我的数据绘制散点图：

fig = plt.figure(1, figsize = (10, 6))
plt.figure()
plt.scatter(X_train_reduced[:, 0],  X_train_reduced[:, 1], c = y_train, cmap = plt.cm.Paired , linewidths=10)

但是它给出了这个错误：

ValueError: 'c' argument must be a mpl color, a sequence of mpl colors or a sequence of numbers, not sample group

当我删除此参数时：

c = y_train

它返回图形，但是所有组都使用相同的颜色（我有2组）。 你知道如何解决吗？