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
Post a Comment