Skip to main content

Read csv file with multiple columns and plot data using matplotlib

Is there an easy way to read a csv file and plot an undefined(csv file changes) number of columns using pandas, matplotlib, and pyqt5? I'm a super beginner to programming so any help would be awesome! I've tried finding articles online to help, but I'm very lost.

The plot I'm trying to create has an x-axis based on time with the format '%d-%m-%Y %H:%M:%S.%f' located in column 0 of the csv file and the y-axis changes depending on the column heading starting with column 1. Example:

Timestamp, O63ME, O63MT, 011KT, etc...

11/06/2020 11:04:00.196, 34, 24, 11, etc...

14/06/2020 11:04:00.449, 114, 29, 3.84, etc...

This is what I have so far:

from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg, NavigationToolbar2QT as Navi
from matplotlib.figure import Figure
import matplotlib.pyplot as plt
import pandas as pd

class MyCanvas(FigureCanvasQTAgg):
    def __init__(self,parent=None, dpi = 120):
        fig = Figure(dpi = dpi)
        self.axes = fig.add_subplot(111)
        super(MyCanvas,self).__init__(fig)
        fig.tight_layout()
    
class MyWindow(QMainWindow):
    def __init__(self):
        QMainWindow.__init__(self)

        self.main_widget = QWidget(self)
        
        """create open button for window"""
        open_button =QPushButton('Open')
        open_button.clicked.connect(self.get_file)
        
        """empty canvas"""
        self.canv = MyCanvas(self)
        
        """create toolbar to control plot"""
        self.toolbar = Navi(self.canv,self.main_widget)
        
        """set layout of window"""
        self.grid = QGridLayout(self.main_widget)
        self.grid.addWidget(open_button, 0,0)
        self.grid.addWidget(self.toolbar,0,1)
        self.grid.addWidget(self.canv,1,0)
        
        self.main_widget.setFocus()
        self.setCentralWidget(self.main_widget)

    def get_file(self):
        getFile(self)
        
class openFunction():
    def __init__(self):
        self.filename = ''
        self.df = []
        
    def getFile(self):
        self.filename = QFileDialog.getOpenFileName(filter = "csv (*.csv)")[0]
        self.readData()
        
    def readData(self):
        self.df = pd.read_csv(instance.filename,encoding = 'utf-8').fillna(0)    

    def plot(self):
        plt.clf()
        plt.plot(df[0:],df[1:])
        

if __name__ == '__main__':

    import sys
    app = QApplication(sys.argv)

    w = MyWindow()
    w.setWindowTitle("Something cool goes here")
    w.show()
    app.exec_()



source https://stackoverflow.com/questions/73872655/read-csv-file-with-multiple-columns-and-plot-data-using-matplotlib

Comments