使用python繪制常用的圖表

更新時(shí)間：2016年08月27日 09:14:11 投稿：hebedich

本文給大家介紹的是如何使用Python根據(jù)Excel表格數(shù)據(jù)繪制不同的圖表的方法，非常的詳細(xì)，有相同需求的小伙伴可以參考下

本文介紹如果使用python匯總常用的圖表，與Excel的點(diǎn)選操作相比，用python繪制圖表顯得比較比較繁瑣，尤其提現(xiàn)在對(duì)原始數(shù)據(jù)的處理上。但兩者在繪制圖表過(guò)程中的思路大致相同，Excel中能完成的工作python大多也能做到。為了更清晰的說(shuō)明使用python繪制圖表的過(guò)程，我們?cè)趨R總圖表的代碼中進(jìn)行注解，說(shuō)明每一行代碼的具體作用。并在文章的最后給出了自定義字體和圖表配色的對(duì)應(yīng)表。

準(zhǔn)備工作

import numpy as np
import pandas as pd
#導(dǎo)入圖表庫(kù)以進(jìn)行圖表繪制
import matplotlib.pyplot as plt
loandata=pd.DataFrame(pd.read_excel('loan_data.xlsx'))

折線(xiàn)圖

#設(shè)置日期字段issue_d為loandata數(shù)據(jù)表索引字段
loandata = loandata.set_index('issue_d')
#按月對(duì)貸款金額loan_amnt求均值，以0填充空值
loan_plot=loandata['loan_amnt'].resample('M').fillna(0)
#圖表字體為華文細(xì)黑，字號(hào)為15
plt.rc('font', family='STXihei', size=15)
#創(chuàng)建一個(gè)一維數(shù)組賦值給a
a=np.array([1,2,3,4,5,6,7,8,9,10,11,12])
#創(chuàng)建折線(xiàn)圖，數(shù)據(jù)源為按月貸款均值，標(biāo)記點(diǎn)，標(biāo)記線(xiàn)樣式，線(xiàn)條寬度，標(biāo)記點(diǎn)顏色和透明度
plt.plot(loan_plot,'g^',loan_plot,'g-',color='#99CC01',linewidth=3,markeredgewidth=3,markeredgecolor='#99CC01',alpha=0.8)
#添加x軸標(biāo)簽
plt.xlabel('月份')
#添加y周標(biāo)簽
plt.ylabel('貸款金額')
#添加圖表標(biāo)題
plt.title('分月貸款金額分布')
#添加圖表網(wǎng)格線(xiàn)，設(shè)置網(wǎng)格線(xiàn)顏色，線(xiàn)形，寬度和透明度
plt.grid( color='#95a5a6',linestyle='--', linewidth=1 ,axis='y',alpha=0.4)
#設(shè)置數(shù)據(jù)分類(lèi)名稱(chēng)
plt.xticks(a, ('1月','2月','3月','4月','5月','6月','7月','8月','9月','10月','11月','12月') )
#輸出圖表
plt.show()

柱狀圖

#按用戶(hù)等級(jí)grade字段對(duì)貸款金額進(jìn)行求和匯總
loan_grade=loandata.groupby('grade')['loan_amnt'].agg(sum)
#圖表字體為華文細(xì)黑，字號(hào)為15
plt.rc('font', family='STXihei', size=15)
#創(chuàng)建一個(gè)一維數(shù)組賦值給a
a=np.array([1,2,3,4,5,6])
#創(chuàng)建柱狀圖，數(shù)據(jù)源為按用戶(hù)等級(jí)匯總的貸款金額，設(shè)置顏色，透明度和外邊框顏色
plt.bar([1,2,3,4,5,6],loan_grade,color='#99CC01',alpha=0.8,align='center',edgecolor='white')
#設(shè)置x軸標(biāo)簽
plt.xlabel('用戶(hù)等級(jí)')
#設(shè)置y周標(biāo)簽
plt.ylabel('貸款金額')
#設(shè)置圖表標(biāo)題
plt.title('不同用戶(hù)等級(jí)的貸款金額分布')
#設(shè)置圖例的文字和在圖表中的位置
plt.legend(['貸款金額'], loc='upper right')
#設(shè)置背景網(wǎng)格線(xiàn)的顏色，樣式，尺寸和透明度
plt.grid(color='#95a5a6',linestyle='--', linewidth=1,axis='y',alpha=0.4)
#設(shè)置數(shù)據(jù)分類(lèi)名稱(chēng)
plt.xticks(a,('A級(jí)','B級(jí)','C級(jí)','D級(jí)','E級(jí)','F級(jí)'))
#顯示圖表
plt.show()

條形圖

#圖表字體為華文細(xì)黑，字號(hào)為15
plt.rc('font', family='STXihei', size=15)
#創(chuàng)建一個(gè)一維數(shù)組賦值給a
a=np.array([1,2,3,4,5,6])
#創(chuàng)建條形圖，數(shù)據(jù)源為分等級(jí)貸款金額匯總，設(shè)置顏色，透明度和圖表邊框
plt.barh([1,2,3,4,5,6],loan_grade,color='#99CC01',alpha=0.8,align='center',edgecolor='white')
#添加x軸標(biāo)題
plt.xlabel('貸款金額')
#添加y軸標(biāo)題
plt.ylabel('用戶(hù)等級(jí)')
#添加圖表標(biāo)題
plt.title('不同用戶(hù)等級(jí)的貸款金額分布')
#添加圖例，并設(shè)置在圖表中的顯示位置
plt.legend(['貸款金額'], loc='upper right')
#設(shè)置背景網(wǎng)格線(xiàn)的顏色，樣式，尺寸和透明度
plt.grid(color='#95a5a6',linestyle='--', linewidth=1,axis='y',alpha=0.4)
#設(shè)置數(shù)據(jù)分類(lèi)名稱(chēng)
plt.yticks(a,('A級(jí)','B級(jí)','C級(jí)','D級(jí)','E級(jí)','F級(jí)'))
#顯示圖表
plt.show()

餅圖

#圖表字體為華文細(xì)黑，字號(hào)為15
plt.rc('font', family='STXihei', size=15)
#設(shè)置餅圖中每個(gè)數(shù)據(jù)分類(lèi)的顏色
colors = ["#99CC01","#FFFF01","#0000FE","#FE0000","#A6A6A6","#D9E021"]
#設(shè)置餅圖中每個(gè)數(shù)據(jù)分類(lèi)的名稱(chēng)
name=['A級(jí)', 'B級(jí)', 'C級(jí)', 'D級(jí)', 'E級(jí)','F級(jí)']
#創(chuàng)建餅圖，設(shè)置分類(lèi)標(biāo)簽，顏色和圖表起始位置等
plt.pie(loan_grade,labels=name,colors=colors,explode=(0, 0, 0.15, 0, 0, 0),startangle=60,autopct='%1.1f%%')
#添加圖表標(biāo)題
plt.title('不同用戶(hù)等級(jí)的貸款金額占比')
#添加圖例，并設(shè)置顯示位置
plt.legend(['A級(jí)','B級(jí)','C級(jí)','D級(jí)','E級(jí)','F級(jí)'], loc='upper left')
#顯示圖表
plt.show()

散點(diǎn)圖

#按月匯總貸款金額，以0填充空值
loan_x=loandata['loan_amnt'].resample('M',how=sum).fillna(0)
#按月匯總利息金額，以0填充空值
loan_y=loandata['total_rec_int'].resample('M',how=sum).fillna(0)
#圖表字體為華文細(xì)黑，字號(hào)為15
plt.rc('font', family='STXihei', size=15)
#創(chuàng)建散點(diǎn)圖，貸款金額為x，利息金額為y，設(shè)置顏色，標(biāo)記點(diǎn)樣式和透明度等
plt.scatter(loan_x,loan_y,60,color='white',marker='o',edgecolors='#0D8ECF',linewidth=3,alpha=0.8)
#添加x軸標(biāo)題
plt.xlabel('貸款金額')
#添加y軸標(biāo)題
plt.ylabel('利息收入')
#添加圖表標(biāo)題
plt.title('貸款金額與利息收入')
#設(shè)置背景網(wǎng)格線(xiàn)的顏色，樣式，尺寸和透明度
plt.grid(color='#95a5a6',linestyle='--', linewidth=1,axis='both',alpha=0.4)
#顯示圖表
plt.show()

氣泡圖

#按月匯總貸款金額及利息
loan_x=loandata['loan_amnt'].resample('M',how=sum).fillna(0)
loan_y=loandata['total_rec_int'].resample('M',how=sum).fillna(0)
loan_z=loandata['total_rec_int'].resample('M',how=sum).fillna(0)
#圖表字體為華文細(xì)黑，字號(hào)為15
plt.rc('font', family='STXihei', size=15)
#設(shè)置氣泡圖顏色
colors = ["#99CC01","#FFFF01","#0000FE","#FE0000","#A6A6A6","#D9E021",'#FFF16E','#0D8ECF','#FA4D3D','#D2D2D2','#FFDE45','#9b59b6']
#創(chuàng)建氣泡圖貸款金額為x，利息金額為y，同時(shí)設(shè)置利息金額為氣泡大小，并設(shè)置顏色透明度等。
plt.scatter(loan_x,loan_y,s=loan_z,color=colors,alpha=0.6)
#添加x軸標(biāo)題
plt.xlabel('貸款金額')
#添加y軸標(biāo)題
plt.ylabel('利息收入')
#添加圖表標(biāo)題
plt.title('貸款金額與利息收入')
#設(shè)置背景網(wǎng)格線(xiàn)的顏色，樣式，尺寸和透明度
plt.grid(color='#95a5a6',linestyle='--', linewidth=1,axis='both',alpha=0.4)
#顯示圖表
plt.show()

箱線(xiàn)圖

#圖表字體為華文細(xì)黑，字號(hào)為15
plt.rc('font', family='STXihei', size=15)
#創(chuàng)建箱線(xiàn)圖，數(shù)據(jù)源為貸款來(lái)源，設(shè)置橫向顯示
plt.boxplot(loandata['loan_amnt'],1,'rs',vert=False)
#添加x軸標(biāo)題
plt.xlabel('貸款金額')
#添加圖表標(biāo)題
plt.title('貸款金額分布')
#設(shè)置背景網(wǎng)格線(xiàn)的顏色，樣式，尺寸和透明度
plt.grid(color='#95a5a6',linestyle='--', linewidth=1,axis='both',alpha=0.4)
#顯示圖表
plt.show()

直方圖

#圖表字體為華文細(xì)黑，字號(hào)為15
plt.rc('font', family='STXihei', size=15)
#創(chuàng)建直方圖，數(shù)據(jù)源為貸款金額，將數(shù)據(jù)分為8等份顯示，設(shè)置顏色和顯示方式，透明度等
plt.hist(loandata['loan_amnt'],8,normed=1, histtype='stepfilled',facecolor='#99CC01', rwidth=0.9,alpha=0.6,edgecolor='white')
#添加x軸標(biāo)題
plt.xlabel('貸款金額')
#添加y軸標(biāo)題
plt.ylabel('概率')
#添加圖表標(biāo)題
plt.title('貸款金額概率密度')
#設(shè)置背景網(wǎng)格線(xiàn)的顏色，樣式，尺寸和透明度
plt.grid(color='#95a5a6',linestyle='--', linewidth=1,axis='y',alpha=0.4)
#顯示圖表
plt.show()