python scatter散點圖用循環(huán)分類法加圖例
更新時間:2019年03月19日 08:39:14 作者:xiaobaicai4552
這篇文章主要為大家詳細介紹了python scatter散點圖用循環(huán)分類法加圖例,具有一定的參考價值,感興趣的小伙伴們可以參考一下
本文實例為大家分享了python scatter散點圖用循環(huán)分類法加圖例,供大家參考,具體內容如下
import matplotlib.pyplot as plt import kNN plt.rcParams['font.sans-serif']=['Simhei'] plt.rcParams['axes.unicode_minus']=False datingDataMat, datingLabels = kNN.file2matrix('datingTestSet2.txt') plt.figure() type1_x = [] #一共有3類,所以定義3個空列表準備接受數(shù)據(jù) type1_y = [] type2_x = [] type2_y = [] type3_x = [] type3_y = [] for i in range(len(datingLabels)): #1000組數(shù)據(jù),i循環(huán)1000次 if datingLabels[i] == '1': #根據(jù)標簽進行數(shù)據(jù)分類,注意標簽此時是字符串 type1_x.append(datingDataMat[i][0]) #取的是樣本數(shù)據(jù)的第一列特征和第二列特征 type1_y.append(datingDataMat[i][1]) if datingLabels[i] == '2': type2_x.append(datingDataMat[i][0]) type2_y.append(datingDataMat[i][1]) if datingLabels[i] == '3': type3_x.append(datingDataMat[i][0]) type3_y.append(datingDataMat[i][1]) plt.scatter(type1_x, type1_y, s=20, c='r', label='不喜歡') plt.scatter(type2_x, type2_y, s=40, c='b', label='魅力一般') plt.scatter(type3_x, type3_y, s=60, c='k', label='極具魅力') plt.legend() plt.show()
用面向對象的寫法:
import matplotlib.pyplot as plt import kNN plt.rcParams['font.sans-serif']=['Simhei'] plt.rcParams['axes.unicode_minus']=False datingDataMat, datingLabels = kNN.file2matrix('datingTestSet2.txt') plt.figure() axes = plt.subplot(111) type1_x = [] type1_y = [] type2_x = [] type2_y = [] type3_x = [] type3_y = [] for i in range(len(datingLabels)): if datingLabels[i] == '1': type1_x.append(datingDataMat[i][0]) type1_y.append(datingDataMat[i][1]) if datingLabels[i] == '2': type2_x.append(datingDataMat[i][0]) type2_y.append(datingDataMat[i][1]) if datingLabels[i] == '3': type3_x.append(datingDataMat[i][0]) type3_y.append(datingDataMat[i][1]) type1 = axes.scatter(type1_x, type1_y, s=20, c='r') type2 = axes.scatter(type2_x, type2_y, s=40, c='b') type3 = axes.scatter(type3_x, type3_y, s=60, c='k') plt.legend((type1, type2, type3), ('不喜歡', '魅力一般', '極具魅力')) plt.show()
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
相關文章
Python利用Matplotlib繪制柱狀圖(豎直柱狀圖和水平柱狀圖)、直方圖和餅狀圖
這篇文章主要給大家介紹了關于Python利用Matplotlib繪制柱狀圖(豎直柱狀圖和水平柱狀圖)、直方圖和餅狀圖的相關資料,Python使用matplotlib畫圖是非常方便的,文中通過代碼介紹的非常詳細,需要的朋友可以參考下2023-12-12幾行代碼讓 Python 函數(shù)執(zhí)行快 30 倍
Python 編程語言,與其他流行編程語言相比主要缺點是它的動態(tài)特性和多功能屬性拖慢了速度表現(xiàn)。Python 代碼是在運行時被解釋的,而不是在編譯時被編譯為原生代碼。在本文中,我們將討論如何用多處理模塊并行執(zhí)行自定義 Python 函數(shù),并進一步對比運行時間指標。2021-10-10