matplotlib与numpy

Posted 2020-10-31 三天泪

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参数：