From c61b4572327225ccce9a138d9f4dcb34a11710ca Mon Sep 17 00:00:00 2001 From: p4iw2ug7a <1744671285@qq.com> Date: Mon, 9 May 2022 10:37:10 +0800 Subject: [PATCH] ADD file via upload --- view.py | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 view.py diff --git a/view.py b/view.py new file mode 100644 index 0000000..fef7795 --- /dev/null +++ b/view.py @@ -0,0 +1,35 @@ +from collections import Counter, defaultdict +import matplotlib +import matplotlib.pyplot as plt +import numpy as np +from sklearn import datasets +import pandas as pd + +iris_datas = datasets.load_iris() +#Iris = pd.read_csv('iris.csv') + + +matplotlib.rcParams['font.sans-serif'] = ['SimHei'] + +style_list = ['o', '^', 's'] # 设置点的不同形状,不同形状默认颜色不同,也可自定义 +data = iris_datas.data +labels = iris_datas.target_names +cc = defaultdict(list) + +for i, d in enumerate(data): + cc[labels[int(i / 50)]].append(d) +p_list = [] +c_list = [] + +for each in [0, 2]: + for i, (c, ds) in enumerate(cc.items()): + draw_data = np.array(ds) + p = plt.plot(draw_data[:, each], draw_data[:, each + 1], style_list[i]) + p_list.append(p) + c_list.append(c) + + plt.legend(map(lambda x: x[0], p_list), c_list) + plt.title('鸢尾花花瓣的长度和宽度') if each else plt.title('鸢尾花花萼的长度和宽度') + plt.xlabel('花瓣的长度(cm)') if each else plt.xlabel('花萼的长度(cm)') + plt.ylabel('花瓣的宽度(cm)') if each else plt.ylabel('花萼的宽度(cm)') + plt.savefig('分布图')