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

Cách thêm legend cho biểu đồ cột Seaborn facetgrid dùng Matplotlib?

Các bước

  • Thiết lập kích thước hình và điều chỉnh khoảng đệm subplot.
  • Tạo DataFrame với cột col1.
  • Tạo lưới đa biểu đồ (FacetGrid) cho mối quan hệ có điều kiện.
  • Dùng map_dataframe() để vẽ với hàm chấp nhận DataFrame dạng dài.
  • Thêm legend cho plot().
  • Hiển thị hình bằng show().

Ví dụ

import pandas as pd
import seaborn as sns
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
df = pd.DataFrame({'col1': [3, 7, 8]})
g = sns.FacetGrid(df, col="col1", hue="col1")
g.map_dataframe(sns.scatterplot)
g.set_axis_labels("X", "Y")
g.add_legend()
plt.show()

Kết quả

Cách thêm legend cho biểu đồ cột Seaborn facetgrid dùng Matplotlib?