How can I have the matplotlib graph with pandas.plot() in the widget?

Question

So far I have this code: When it's executed, you'll see two windows main and another one called Figure 1, when the csv data is readed, and the plot button pressed, The graph is shown in another window (Figure 2) How can I fix this issues? Many, many thanks in advance for your time!

import sys
import pandas as pd
import matplotlib
matplotlib.use('QT5Agg')
from PyQt5 import QtWidgets
from PyQt5.QtWidgets import QDialog, QApplication, QPushButton, QVBoxLayout
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar
import matplotlib.pyplot as plt
from matplotlib import style
style.use('ggplot')


class Window(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setGeometry(600,300, 1000, 600)
        self.center()
        grid = QtWidgets.QGridLayout()
        widget = QtWidgets.QWidget(self)
        self.setCentralWidget(widget)
        widget.setLayout(grid)
         #Import CSV Button
        btn1 = QtWidgets.QPushButton('Import CSV', self)
        btn1.resize(btn1.sizeHint())
        btn1.clicked.connect(self.getCSV)
        grid.addWidget(btn1, 1, 0)

        self.figure = plt.figure(figsize=(15,5))    
        self.canvas = FigureCanvas(self.figure)     
        grid.addWidget(self.canvas, 2,0,1,2)


        btn2 = QtWidgets.QPushButton('Plot', self)
        btn2.resize(btn2.sizeHint())    
        btn2.clicked.connect(self.plot)
        grid.addWidget(btn2, 1, 1)
        self.show()

    def plot(self):
        self.df.plot(x='col1',y='col2')
        self.canvas.draw()

    def getCSV(self):
         filePath, _ = QtWidgets.QFileDialog.getOpenFileName(self, 'Open file', '/home')
         if filePath != "":
            print ("Dirección",filePath)
            self.df = pd.read_csv(str(filePath))

    def center(self):
        qr = self.frameGeometry()
        cp = QtWidgets.QDesktopWidget().availableGeometry().center()
        qr.moveCenter(cp)
        self.move(qr.topLeft())

def main():
    app = QtWidgets.QApplication(sys.argv)
    w = Window()
    sys.exit(app.exec_())


if __name__ == '__main__':
   main()

eyllanesc · Accepted Answer · 2017-12-25 23:25:55Z

1

the plot() function of pandas has an optional attribute called ax:

ax : matplotlib axes object, default None

If we want it to be drawn inside FigureCanvas() you must first create some axes, and pass it as an attribute in ax:

def plot(self):
    self.figure.clear()
    ax =  self.figure.add_subplot(111) 
    self.df.plot(x='col1',y='col2', ax=ax)
    self.canvas.draw()

answered Dec 25, 2017 at 23:25

eyllanesc

246k19 gold badges205 silver badges282 bronze badges

Sign up to request clarification or add additional context in comments.

6 Comments

Liliana Castañón Over a year ago

Thanks again for your help! Worked very nice! Any ideas to get rid of the Figure 1, that appears at the same time as the main window?

eyllanesc Over a year ago

I do not understand you, could you explain to me what you want to get? :P

Liliana Castañón Over a year ago

Gladly :D ! When I execute the code, I'm getting two windows (here the screenshot: drive.google.com/file/d/163lRSxyApUAwRJEmdSaQOBIDnt1oLANS/… )

eyllanesc Over a year ago

I find it strange that this happens to you, I could help you through teamviewer since that error does not occur on my pc.

Liliana Castañón Over a year ago

I'm using Ubuntu and Spyder, maybe something there is generating the error, let me see if I can fix it ^_^ Thanks!

|

Collectives™ on Stack Overflow

How can I have the matplotlib graph with pandas.plot() in the widget?

1 Answer 1

6 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

6 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related