matplotlib与numpy

import matplotlib.pyplot as plt
import numpy as np

y = df.loc[0:100, 4].values
y = np.where(y == ‘Iris-setosa‘,-1,1)
X = df.iloc[0:100,[0,2]].values

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()

代码补寄：

plt.scatter中color参数：

‘b’          blue             蓝色

‘g’          green           绿色

‘r’          red               红色

‘c’          cyan             青色

‘y’          yellow           黄色

‘k’          black            黑色

‘w’          white            白色

‘m’          magenta      品红

marker参数：

matplotlib与numpy

原文：https://www.cnblogs.com/3daytears/p/8890412.html