Matplotlab可視化學習筆記(二):如何繪制柱狀圖

文章目錄

• • 1.安裝matplotlib
  • 2.具體代碼

1.安裝matplotlib

matplotlib是一個非常實用的畫柱狀圖的tool，通過pip install matplotlib直接安裝即可

2.具體代碼

import matplotlib.pyplot as plt
import inspect
yolov4_time,yolo_time,ssdface_time, libface_time=0.1,0.5,0.3,0.2
waste_times = (yolov4_time,yolo_time,ssdface_time, libface_time)

labels = []
def retrieve_name(var):
    callers_local_vars = inspect.currentframe().f_back.f_locals.items()
    return [var_name for var_name, var_val in callers_local_vars if var_val is var]

for data in waste_times:
    labels.append(retrieve_name(data)[0].replace('_time', ''))

plt.rcParams['font.family'] = 'SimHei'  # 設定中文解決matplotlib中文亂碼問題
x = list(range(len(waste_times)))
plt.bar(x, waste_times, width=0.5, color='red', label='耗時比較', tick_label=labels)
for a, b in zip(x, waste_times):
    plt.text(a, b, '%.1f' % b, ha='center', va='bottom', fontsize=10)  # 添加資料标簽

plt.xlabel("模型")
plt.ylabel("時間")


plt.legend()
plt.show()
           

運作結果：