如何在matplotlib中获取条形图/堆积条形图上的标签?

我在PyQt Canvas中嵌入了一个非常简单的堆叠matplotlib条形图.我试图根据点击获得条形区域(矩形)的相应标签.但是当我尝试从事件中打印信息时,我总是会得到_nolegend_.理想情况下,我希望在代码中附加条形图上看到相应的标签.

例如,当您单击灰色条时,它应该打印a2

import sys
import matplotlib.pyplot as plt

from PyQt4 import QtGui
from matplotlib.figure import Figure
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas


def on_pick(event):
    print event.artist.get_label()

def main():

    app = QtGui.QApplication(sys.argv)

    w = QtGui.QWidget()
    w.resize(640, 480)
    w.setWindowTitle('Pick Test')

    fig = Figure((10.0, 5.0), dpi=100)
    canvas = FigureCanvas(fig)
    canvas.setParent(w)

    axes = fig.add_subplot(111)

    # bind the pick event for canvas
    fig.canvas.mpl_connect('pick_event', on_pick)

    p1 = axes.bar(1,6,picker=2,label='a1')
    p2 = axes.bar(1,2, bottom=6,color='gray',picker=1,label='a2')

    axes.set_ylim(0,10)
    axes.set_xlim(0,5)

    w.show()

    sys.exit(app.exec_())


if __name__ == '__main__':
    main()

最佳答案 这有点棘手,因为bar是一个复杂的绘图对象,它实际上由多个组件组成.

您可以使用get_legend_handles_labels获取轴的所有艺术家和标签.然后您可以查看当前艺术家所属的组.

所以你的回调可能看起来像这样.

def on_pick(event)
    rect = event.artist

    # Get the artists and the labels
    handles,labels = rect.axes.get_legend_handles_labels()

    # Search for your current artist within all plot groups
    label = [label for h,label in zip(handles, labels) if rect in h.get_children()]

    # Should only be one entry but just double check
    if len(label) == 1:
        label = label[0]
    else:
        label = None

    print label
点赞