更改所选matplotlib直方图箱栏的颜色,给定值
问题描述
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()
退货:
相关文章