更改所选matplotlib直方图bin bar的颜色,给定它的值

2024-06-30 15:51:52 发布

您现在位置:Python中文网/ 问答频道 /正文

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变量自动将值所在直方图上的条形图颜色更改为蓝色,例如:

enter image description here

我怎样才能做到这一点?


Tags: toimportpandasmatplotlibvalueasnpbar
1条回答
网友
1楼 · 发布于 2024-06-30 15:51:52

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

返回:

enter image description here

相关问题 更多 >