#注意看我y和X這兩行的注釋：要么全是df.loc,要么全是df.iloc，核對一下自己代碼

import matplotlib.pyplot as plt
import numpy as np

y = df.loc[0:100, 4].values #loc/iloc得統一
y = np.where(y == 'Iris-setosa', -1, 1)

#抽出第0列和第2列
X = df.loc[0:100, [0, 2]].values # loc/iloc得統一
plt.scatter(X[:50, 0], X[:50, 1], color='red', marker='o', label='setosa')
plt.scatter(X[50:100, 0], X[50:100, 1], color='blue', marker='x', label='versicolor')
plt.xlabel('花瓣長度')
plt.ylabel('花莖長度')
plt.legend(loc='upper left')
#plt.show()