pyqtgraph:为图中的线添加图例

时间:2013-05-22 17:27:43

标签: python pyqtgraph

我正在使用pyqtgraph,我想在InfiniteLines的图例中添加一个项目。

我已经调整了示例代码来演示:

# -*- coding: utf-8 -*-
"""
Demonstrates basic use of LegendItem

"""
import initExample ## Add path to library (just for examples; you do not need this)

import pyqtgraph as pg
from pyqtgraph.Qt import QtCore, QtGui

plt = pg.plot()
plt.setWindowTitle('pyqtgraph example: Legend')
plt.addLegend()

c1 = plt.plot([1,3,2,4], pen='r', name='red plot')
c2 = plt.plot([2,1,4,3], pen='g', fillLevel=0, fillBrush=(255,255,255,30), name='green plot')
c3 = plt.addLine(y=4, pen='y')
# TODO: add legend item indicating "maximum value"

## Start Qt event loop unless running in interactive mode or using pyside.
if __name__ == '__main__':
    import sys
    if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
        QtGui.QApplication.instance().exec_()

我得到的结果是: plot image

如何添加合适的图例项?

2 个答案:

答案 0 :(得分:7)

对于此示例,您可以使用正确的颜色创建一个空的PlotDataItem,并将其添加到图例中,如下所示:

style = pg.PlotDataItem(pen='y')
plt.plotItem.legend.addItem(l, "maximum value")

答案 1 :(得分:7)

如果使用“name”参数创建了项目,则pyqtgraph会自动将项目添加到图例中。上述代码中唯一需要的调整如下:

c3 = plt.plot (y=4, pen='y', name="maximum value")

只要为pyqtgraph提供曲线名称,它就会自动创建相应的图例项目。 尽管在创建曲线之前调用plt.addLegend()非常重要。