pyqt_data_analysis/data_set_read.py

import pandas as pd
import numpy as np

class Data_Read:
    def __init__(self, file_path, suffix) -> None:
        self.file_path = file_path
        if suffix == "csv":
            self.data_frame = pd.read_csv(self.file_path)
        elif suffix == "xls" or suffix == "xlsx":
            self.data_frame = pd.read_excel(self.file_path)
            
    def get_head_data(self, n : int):
        return self.data_frame.head(n)