python畫圖時給圖中的點加標簽和plt.text的使用

Posted on by

背景：

今天在用matplotlib模塊畫各城市2019-nCoV疫情確診人數和節前流入人口數的圖的時候遇到瞭要給圖中的點加上標簽示意，原本圖長這個樣子

現在要給各散點標註是哪個哪個城市，即下面這種圖：

matplotlib模塊加標簽主要有matplotlib.pyplot.text()和matplotlib.pyplot.annotate()兩個關鍵函數，後者適用范圍更廣，今天主要談一下前者matplotlib.pyplot.text()，簡寫成plt.text() 。

準備知識：

在此重點講一下plt.text()的用法和參數設置，plt.text()函數基本語法如下：

plt.text(x, y, s, fontsize, verticalalignment,horizontalalignment,rotation , **kwargs)

其中：

  • x,y表示標簽添加的位置，默認是根據坐標軸的數據來度量的，是絕對值，也就是說圖中點所在位置的對應的值，特別的，如果你要變換坐標系的話，要用到transform=ax.transAxes參數。
  • s表示標簽的符號，字符串格式，比如你想加個“我愛三行科創”，更多的是你標註跟數據有關的主體，你如實寫便是。
  • fontsize顧名思義就是你加標簽字體大小瞭，取整數。
  • verticalalignment表示垂直對齊方式 ，可選 ‘center’ ，‘top’ ， ‘bottom’，‘baseline’ 等
  • horizontalalignment表示水平對齊方式 ，可以填 ‘center’ ， ‘right’ ，‘left’ 等
  • rotation表示標簽的旋轉角度，以逆時針計算，取整
  • 後面還有 family 用來設置字體，style 設置字體的風格，weight 字體的粗細, bbox 給字體添加框,如 bbox=dict(facecolor=‘red’, alpha=0.5) 等，各種風格，應有盡有，總有一款適合你。

實例操作：

講解玩關鍵函數類plt.text()的用法之後，我們要進行實際操作。首先，我們用下面代碼把第一張圖畫出來

import pandas as pd #導入數據分析模塊
import matplotlib.pyplot as plt #導入繪圖模塊類
plt.rcParams['font.sans-serif']=['SimHei'] #圖中文字體設置為黑體
plt.rcParams['axes.unicode_minus']=False #負值顯示
data=pd.read_excel(r"D:\城市租賃地圖\人流流動與疫情發展\百度遷徙.xlsx") #讀取數據
city_name=data['city'] #城市名稱
people_flow=data['out_people']*100 #流出人口,單位百人
confirm=data['confirm(2.10)'] #確診人數

fig=plt.figure(figsize=(8,6)) #新建畫佈
ax=plt.subplot(1,1,1) #子圖初始化
ax.scatter(people_flow,confirm) #繪制散點圖   
ax.set_title("人口流入-確診人數")
ax.set_xlabel("人口流入數（百人）")
ax.set_ylabel("確診人數")
plt.show()

然後再給第一個點加上標簽，隻需要添加一行代碼即可：

ax.text(430, 337, "北京", fontsize=12, color = "r", style = "italic", weight = "light", verticalalignment='center', horizontalalignment='right', rotation=90)

其中430, 337就是這個點的坐標值，北京是這個點的主體，你可以不斷的去調後面的參數使其滿足你所需。

接著，我們要批量給圖中的點加上主體標簽，使其看起來像第二張圖，需要用到循環語句來控制加標簽的位置

for i in range(len(confirm)):
    ax.text(people_flow[i]*1.01, confirm[i]*1.01, city_name[i], fontsize=10, color = "r", style = "italic", weight = "light", verticalalignment='center', horizontalalignment='right',rotation=0) #給散點加標簽

這裡關鍵是裡面偏移參數的設定，比如這裡兩個偏移率都設置成1.01, 當初想的是標簽不要覆蓋原來的點，兩者最好不重合，看起來大方得體，你可以開始的設置成1.0，觀察效果然後慢慢調節，舉個例子，如果是柱狀圖，且寬度適合的話，第一個偏移率設置成1.0，第二個設置成1.05左右，即稍微高出柱子多一丁點顯示柱子的標簽。

完整代碼：

最後給出完整代碼，如果你不會寫代碼或者隻要數據測試，百度遷徙.xlsx如下:

import pandas as pd #導入數據分析模塊
import matplotlib.pyplot as plt #導入繪圖模塊類
plt.rcParams['font.sans-serif']=['SimHei'] #圖中文字體設置為黑體
plt.rcParams['axes.unicode_minus']=False #負值顯示
data=pd.read_excel(r"D:\城市租賃地圖\人流流動與疫情發展\百度遷徙.xlsx") #讀取數據
city_name=data['city'] #城市名稱
people_flow=data['out_people']*100 #流出人口,單位百人
confirm=data['confirm(2.10)'] #確診人數

fig=plt.figure(figsize=(8,6)) #新建畫佈
ax=plt.subplot(1,1,1) #子圖初始化
ax.scatter(people_flow,confirm) #繪制散點圖   
ax.set_title("人口流入-確診人數")
ax.set_xlabel("人口流入數（百人）")
ax.set_ylabel("確診人數")
#ax.text(430, 337, "北京", fontsize=12, color = "r", style = "italic", weight = "light", verticalalignment='center', horizontalalignment='right',rotation=90)
for i in range(len(confirm)):
    ax.text(people_flow[i]*1.01, confirm[i]*1.01, city_name[i], 
            fontsize=10, color = "r", style = "italic", weight = "light",
            verticalalignment='center', horizontalalignment='right',rotation=0) #給散點加標簽
plt.show()

到此這篇關於python畫圖時給圖中的點加標簽和plt.text的使用的文章就介紹到這瞭,更多相關python給圖中點加標簽內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：