Построение графика рассеяния с использованием matplotlib для y = 0 и y = 1 с разными цветами

Question

Я пытался построить график рассеяния, используя matplotlib, где моя ось x и ось y представляют собой две особенности X [0], X [1], и он должен отображать точки с разными цветами, где y = 0 и y =1.

import pandas as pd
import matplotlib.pyplot as plt

df=pd.read_csv('F:/ex2data2.txt', sep=",", header=None)
X=df.iloc[:, :2]
y=df.iloc[:, 2]

def plotData(X,y):
    fig=plt.figure()
    pos = y==1
    neg = y==0
    plt.plot(X[pos, 0], X[pos, 1], 'k*')
    plt.plot(X[neg, 0], X[neg, 1], 'ko')
    plt.show()

plotData(X,y)

Строка "plt.plot (X [pos, 0], X [pos, 1], 'k *') и plt.plot (X [neg, 0], X"[neg, 1], 'ko') "показывает ошибку. Кто-нибудь может мне помочь с причиной?