python - Matplotlib:从条形图中获取颜色和 x/y 数据

Question

我有一个条形图，我想获取它的颜色和 x/y 值。这是一个示例代码:

import matplotlib.pyplot as plt
def main():
    x_values = [1,2,3,4,5]
    y_values_1 = [1,2,3,4,5]
    y_values_2 = [2,4,6,8,10]
    f, ax = plt.subplots(1,1)
    ax.bar(x_values,y_values_2,color='r')
    ax.bar(x_values,y_values_1,color='b')
    #Any methods?
    plt.show()
if __name__ == '__main__':
    main()

有没有什么方法像 ax.get_xvalues(), ax.get_yvalues(), ax.get_colors()，我可以使用它，这样我就可以从 ax 中提取列表 x_values、y_values_1、y_values_2以及颜色 'r' 和 'b'?

Answer 1

ax 知道它正在绘制什么几何对象，但没有记录这些几何对象是何时添加的，当然它也不知道它们是什么“意思”:哪个补丁来自哪个条形图，等等。编码器需要跟踪它以重新提取正确的部分以供进一步使用。许多 Python 程序都采用这种方法:调用 barplot 返回一个 BarContainer，您可以在当时命名并稍后使用:

import matplotlib.pyplot as plt
def main():
    x_values = [1,2,3,4,5]
    y_values_1 = [1,2,3,4,5]
    y_values_2 = [2,4,6,8,10]
    f, ax = plt.subplots(1,1)
    rbar = ax.bar(x_values,y_values_2,color='r')
    bbar = ax.bar(x_values,y_values_1,color='b')
    return rbar, bbar

if __name__ == '__main__':
    rbar, bbar = main()
    # do stuff with the barplot data:
    assert(rbar.patches[0].get_facecolor()==(1.0,0.,0.,1.)) 
    assert(rbar.patches[0].get_height()==2)