Matplotlib：自动识别柱状图中数量Top2的条形并设置颜色

没问题，我来帮你搞定这个需求！核心思路是先动态识别出数量排名前两位的水果，再给对应的条形单独设置红色，其余保持灰色。下面是修改后的完整可运行代码：

import pandas as pd
import matplotlib.pyplot as plt

data = [['Apple',10],['Banana',15],['Kiwi',11],['Orange',17]]
df = pd.DataFrame(data,columns=['Fruit','Quantity'])
df.set_index('Fruit', inplace=True)

# 动态获取数量Top2的水果索引
top2_fruits = df['Quantity'].nlargest(2).index

# 生成对应颜色列表：Top2设为红色，其余灰色
bar_colors = ['red' if fruit in top2_fruits else 'gray' for fruit in df.index]

# 绘制柱状图并传入颜色列表
df.plot.bar(color=bar_colors, rot=0)
plt.show()

关键步骤解释： #

• 动态识别Top2：用df['Quantity'].nlargest(2).index直接提取数量排名前两位的水果名称，完全不需要固定阈值，不管后续数据怎么变化（比如某水果数量更新），都能自动定位当前的Top2。
• 生成颜色映射：通过列表推导式遍历所有水果，判断每个水果是否在Top2集合里，对应设置红色或灰色，这样颜色列表和条形的顺序完全匹配。
• 传入颜色参数：把生成的bar_colors传给plot.bar()的color参数，就能实现预期的颜色效果。

这个方案的优势是完全自适应数据变化，不管你的水果数量、数值怎么调整，都能准确标记出当前数量最高的两个条形~

内容的提问来源于stack exchange，提问作者Saad