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

Tạo biểu đồ Mosaic (Mosaic Plot) trong Matplotlib

Để tạo biểu đồ Mosaic (Mosaic Plot) trong Matplotlib, bạn có thể thực hiện theo các bước sau:

Các bước thực hiện

  • Thiết lập kích thước hình vẽ và điều chỉnh khoảng cách giữa các subplot.
  • Cài đặt gói statsmodels (pip install statsmodels). Gói này cung cấp hàm mosaic để vẽ biểu đồ mosaic. statsmodels là gói Python bổ sung cho scipy cho các tính toán thống kê.
  • Tạo từ điển (dictionary) dữ liệu cho biểu đồ mosaic.
  • Tạo biểu đồ mosaic từ bảng liên eventualities (contingency table).
  • Sử dụng phương thức show() để hiển thị hình vẽ.

Ví dụ minh họa

import matplotlib.pyplot as plt
from statsmodels.graphics.mosaicplot import mosaic

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

# Dữ liệu cho mosaic plot
data = {'John': 7, 'Joe': 10, 'James': 5, 'Kate': 1}

# Tạo mosaic plot
mosaic(data, title='Basic mosaic plot')

plt.show()

Kết quả đầu ra

Biểu đồ mosaic hiển thị tỷ lệ tần suất của 4 danh mục (John, Joe, James, Kate) bằng diện tích hình chữ nhật tương ứng.