Làm thế nào để thay đổi màu của một thanh đơn nếu một điều kiện là đúng (Matplotlib)?

Để thay đổi màu của một thanh đơn nếu một điều kiện là đúng, chúng ta có thể tạo một tập hợp các giá trị và danh sách các màu với màu đỏ cho đến khi giá trị là 2; nếu không, hãy thêm màu vàng vào danh sách.

Các bước

• Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô phụ.
• Khởi tạo chiều rộng thay đổi của thanh.
• Lập hai danh sách các giá trị và màu sắc.
• Sử dụng bar () phương pháp để vẽ các thanh.
• Để hiển thị hình này, hãy sử dụng show () phương pháp.

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()

Đầu ra