Ошибка при построении матрицы проекции в Matlablib

Question

Я пытаюсь выполнить следующий код Python для построения матрицы проекции с набором данных wine, но я получаю ошибку. Можете ли вы помочь мне понять это?

Код:

with plt.style.context('seaborn-whitegrid'):

    plt.figure(figsize=(6, 4))
    for lab, col in zip(('Class_1', 'Class_2', 'Class_3'),
                        ('blue', 'red', 'green')):
        plt.scatter(X_train_pca[Y_train==lab, 0],
                    X_train_pca[Y_train==lab, 1],
                    label=lab,
                    c=col)
    plt.xlabel('Principal Component 1')
    plt.ylabel('Principal Component 2')
    plt.legend(loc='lower center')
    plt.tight_layout()
    plt.show()

Ошибка:

TypeError Traceback (most recent call last)
<ipython-input-21-eeec1ead5b41> in <module>
----> 5         plt.scatter(X_train_pca[Y_train==lab, 0], X_train_pca[Y_train==lab, 1],label=lab,

  47   False
  172  False

  [123 rows x 1 columns], 0)' is an invalid key

  &lt;Figure size 432x288 with 0 Axes&gt;