import matplotlib.pyplot as plt
import pandas as pd
# 读取csv信息
data_csv = pd.read_csv('iris.csv')
# 提取花的类型
flowers_type = list(set(data_csv['species'].values))
# 将同种类型的花提取出来,然后放入列表
temp = []
for i in flowers_type:
temp.append(data_csv.loc[data_csv['species'] == i])
# 设置中文显示
plt.rcParams['font.sans-serif'] = ['SIMHEI']
# 设置图片大小
plt.figure(figsize=(15,6))
# 循环将各种花的数据导入图表中
for i in range(len(temp)):
sepal_length = temp[i]['sepal_length']
sepal_width = temp[i]['sepal_width']
petal_length = temp[i]['petal_length']
petal_width = temp[i]['petal_width']
# 1*2 的图片网格的第一个位置的图片
plt.subplot(121)
# 将花瓣长度&花瓣宽度导入图表
plt.scatter(sepal_length,sepal_width,label=flowers_type[i])
# 1*2 的图片网格的第二个位置的图片
plt.subplot(122)
# 将花萼长度&花萼宽度导入图表
plt.scatter(petal_length,petal_width,label=flowers_type[i])
# 设置第一个图的标题、x、y轴信息
plt.subplot(121)
plt.title('花瓣长度&花瓣宽度散点图')
plt.ylabel('花瓣长度')
plt.xlabel('花瓣宽度')
# 显示网格 透明度为0.3
plt.grid(alpha=0.3)
# 显示图例
plt.legend()
# 设置第二个图的标题、x、y轴信息
plt.subplot(122)
plt.title('花萼长度&花萼宽度散点图')
plt.ylabel('花萼长度')
plt.xlabel('花萼宽度')
# 显示网格 透明度为0.3
plt.grid(alpha=0.3)
# 显示图例
plt.legend()
# 显示图片
plt.show()