【matplotlib】数字がプラスなら青、マイナスなら赤の棒グラフを描きたい【python】

会計関連のグラフを作るときに、プラスとマイナスで色を変えたくなった。

どうやるのかと思って調べてみたところ、それ用のリストを作成するのがいいらしい。

 

import matplotlib.pyplot as plt

# データの例
values = [3, 2, -2.3, 1.2, -0.4]
labels = ['A', 'B', 'C', 'D', 'E']

# 棒グラフを描画する際に、プラスの値は青、マイナスの値は赤にする
colors = ['blue' if x >= 0 else 'red' for x in values]

plt.figure(figsize=(10, 6))
plt.bar(labels, values, color=colors)
plt.show()