更改所选matplotlib直方图箱栏的颜色,给定值

2022-03-04 00:00:00 python matplotlib plot bar-chart

问题描述

Similar to a question I asked previously,我有一个这样的MWE:

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np

pd.Series(np.random.normal(0, 100, 1000)).plot(kind='hist', bins=50, color='orange')

bar_value_to_colour = 102

然后我想使用bar_value_to_colour变量自动将值所在的直方图上的条形图颜色更改为蓝色,例如:

如何实现此目标?


解决方案

使用rectangle.get_x()很容易获得条形图的x坐标,但问题是条形图不是精确绘制在特定值上,所以我必须选择最接近的一个。以下是我的解决方案:

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

s = pd.Series(np.random.normal(0, 100, 10000))
p = s.plot(kind='hist', bins=50, color='orange')

bar_value_to_label = 100
min_distance = float("inf")  # initialize min_distance with infinity
index_of_bar_to_label = 0
for i, rectangle in enumerate(p.patches):  # iterate over every bar
    tmp = abs(  # tmp = distance from middle of the bar to bar_value_to_label
        (rectangle.get_x() +
            (rectangle.get_width() * (1 / 2))) - bar_value_to_label)
    if tmp < min_distance:  # we are searching for the bar with x cordinate
                            # closest to bar_value_to_label
        min_distance = tmp
        index_of_bar_to_label = i
p.patches[index_of_bar_to_label].set_color('b')

plt.show()

退货:

相关文章