用Matplotlib画一条垂直线

时间:2015-03-17 10:38:25

标签: qt matplotlib

我想用Matpotlib绘制一条垂直线。我发现这种方法:axvline但它不起作用。 这是我的代码:

import sys
import matplotlib
matplotlib.use('Qt4Agg')

from ui_courbe import *
from PyQt4 import  QtGui
from matplotlib import pyplot as plt


class Window(QtGui.QMainWindow, Ui_MainWindow):
    def __init__(self, parent=None):
        super(Window, self).__init__(parent)
        self.setupUi(self) 
        self.boutonDessiner.clicked.connect(self.generatePlot)

    def generatePlot(self):
        # generate the plot 
        ax = self.graphicsView.canvas.fig.add_subplot(111)
        ax.plot([1,3,5,7],[2,5,1,-2])
        plt.axvline(x=4)
        self.graphicsView.canvas.draw()


if __name__ == '__main__':
    app = QtGui.QApplication(sys.argv)
    win = Window()
    win.show()
    sys.exit(app.exec_())

我可以看到我的情节,但没有垂直线。为什么呢?

致以最诚挚的问候,

菲利普

2 个答案:

答案 0 :(得分:5)

您的示例不是自包含的,但我认为您需要替换:

plt.axvline(x=4)

使用:

ax.axvline(x=4)

您正在将线条添加到未显示的轴。使用plt.是您可能希望避免GUI的pyplot接口。所以你所有的绘图都必须放在像ax.

这样的轴上

答案 1 :(得分:1)

matplotlib.pyplot.vlines

  • 区别在于您可以将x多个位置作为列表传递,而matplotlib.pyplot.axvline仅允许一个位置。
    • 单个位置:x=37
    • 多个位置:x=[37, 38, 39]
  • 如果要用fig, ax = plt.subplots()之类的图形来绘制图形,则分别用plt.vlinesplt.axvline替换ax.vlinesax.axvline
import numpy as np
import matplotlib.pyplot as plt

xs = np.linspace(1, 21, 200)
plt.vlines(x=[37, 38, 39], ymin=0, ymax=len(xs), colors='purple', ls='--', lw=2, label='vline_multiple')
plt.vlines(x=40, ymin=0, ymax=len(xs), colors='green', ls=':', lw=2, label='vline_single')
plt.axvline(x=36, color='b', label='avline')
plt.legend()
plt.show()

enter image description here