Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีเปลี่ยนสีแท่งเดี่ยวเมื่อเงื่อนไขเป็นจริงใน Matplotlib


สร้างชุดค่าและรายการสี สีแดงจนกว่าค่าจะเป็น 2 ไม่ใช่ให้เพิ่มสีเหลือง

ขั้นตอนการทำงาน

  • กำหนดขนาดรูปภาพและปรับระยะห่างระหว่างและรอบๆ subplots
  • กำหนดตัวแปร width ของแท่ง
  • สร้างสองรายการค่าและสี
  • ใช้ bar() พล็อตแท่ง
  • ใช้เมธอด show() เพื่อแสดงรูปภาพ

ตัวอย่างโค้ด

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

ผลลัพธ์

วิธีเปลี่ยนสีแท่งเดี่ยวเมื่อเงื่อนไขเป็นจริงใน Matplotlib