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ù)據
type1_y = []
type2_x = []
type2_y = []
type3_x = []
type3_y = []
for i in range(len(datingLabels)): #1000組數(shù)據，i循環(huán)1000次
if datingLabels[i] == '1': #根據標簽進行數(shù)據分類,注意標簽此時是字符串
type1_x.append(datingDataMat[i][0]) #取的是樣本數(shù)據的第一列特征和第二列特征
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中使用xlrd、xlwt操作excel表格詳解,python操作excel主要用到xlrd和xlwt這兩個庫，即xlrd是讀excel，xlwt是寫excel的庫,需要的朋友可以參考下2015-01-01
今天小編就為大家分享一篇使用python 將圖片復制到系統(tǒng)剪貼中，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
python之文件的讀寫和文件目錄以及文件夾的操作實現(xiàn)代碼
這篇文章主要介紹了python之文件的讀寫和文件目錄以及文件夾的操作實現(xiàn)代碼,需要的朋友可以參考下2016-08-08
這篇文章主要為大家詳細介紹了Python使用tkinter制作在線翻譯軟件，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-02-02
這篇文章主要介紹了Python多線程編程之threading模塊詳解,文中有非常詳細的代碼示例,對正在學習python的小伙伴們有非常好的幫助,需要的朋友可以參考下2021-04-04
這篇文章主要給大家介紹了關于python中pop()函數(shù)語法與實例的相關資料，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-12-12
Python實現(xiàn)圖片指定位置加圖片水?。ǜ絇yinstaller打包exe)
這篇文章主要介紹了Python實現(xiàn)圖片指定位置加圖片水印,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-03-03
在 Python 中,下劃線（underscore）有多種用法,它在不同的上下文中可以扮演不同的角色,本文將介紹python中的下劃線用法總結,感興趣的朋友一起看看吧2024-05-05
詳解Python中表達式i += x與i = i + x是否等價
這篇文章主要介紹了關于Python中表達式i += x與i = i + x是否等價的相關資料，文中通過示例代碼介紹的很詳細，相信對大家具有一定的參考價值，有需要的朋友們下面來一起看看吧。2017-02-02
python實現(xiàn)優(yōu)雅的打印json格式文本
這篇文章主要介紹了python實現(xiàn)優(yōu)雅的打印json格式文本方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-08-08