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

Đổi màu một cột đơn lẻ theo điều kiện trong Matplotlib

Để tô màu khác cho cột thỏa mãn điều kiện (ví dụ giá trị = 2), tạo list màu tương ứng từng cột trước khi vẽ.

Các bước

  • Thiết lập figure size và autolayout.
  • Khởi tạo width, mảng giá trị vals.
  • Tạo list màu: ["red" if i != 2 else "yellow" for i in vals].
  • Vẽ bar chart với color=colors.
  • Hiển thị bằng show().

Ví dụ

import numpy as np
import matplotlib.pyplot as plt

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

data = np.arange(5)
width = 0.5
vals = [1, 2, 1, 5, 3]
colors = ["red" if i != 2 else "yellow" for i in vals]
plt.bar(data, vals, width, color=colors)
plt.show()

Kết quả

Đổi màu một cột đơn lẻ theo điều kiện trong Matplotlib