Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Tạo lollipop plot đơn giản với Matplotlib

Để tạo lollipop plot (biểu đồ kẹo mút) đơn giản trong Matplotlib:

  • Thiết lập kích thước figure và padding
  • Tạo DataFrame df dữ liệu 2 chiều
  • Sắp xếp DataFrame bằng sort_values()
  • Tạo list range theo index DataFrame
  • Vẽ stem plot từ DataFrame đã sắp xếp
  • Đặt xticks và labels bằng xticks()
  • Hiển thị bằng show()

Ví dụ mã nguồn

import numpy as np
from matplotlib import pyplot as plt
import pandas as pd

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

df = pd.DataFrame({'group': list(map(chr, range(65, 85))), 'values': np.random.uniform(size=20)})
ordered_df = df.sort_values(by='values')
my_range = range(1, len(df.index) + 1)
plt.stem(ordered_df['values'])
plt.xticks(my_range, ordered_df['group'])
plt.show()

Kết quả

Biểu đồ lollipop: các đường thẳng đứng (stems) từ trục X đến giá trị, đầu có marker tròn, sắp xếp tăng dần.

Tạo lollipop plot đơn giản với Matplotlib

Đặc điểm: Lollipop plot = stem plot + sắp xếp giá trị. Thích hợp so sánh thứ hạng các nhóm.