Parse CSV Data with a Python Class

class DataParser:
    def __init__(self, file_path):
        self.file_path = file_path
        self.data = []

    def load_data(self):
        with open(self.file_path, 'r') as file:
            for line in file:
                row = line.strip().split(',')
                self.data.append(row)
        return self.data

    def get_column(self, column_index):
        return [row[column_index] for row in self.data]

    def get_row(self, row_index):
        return self.data[row_index]

    def row_count(self):
        return len(self.data)

    def column_count(self):
        return len(self.data[0]) if self.data else 0


if __name__ == "__main__":
    sample_data = "name,age,city\nAlice,25,New York\nBob,30,Los Angeles\nCharlie,35,Chicago"
    with open('sample_data.csv', 'w') as f:
        f.write(sample_data)

    parser = DataParser('sample_data.csv')
    parser.load_data()
    
    print(f"Rows: {parser.row_count()}, Columns: {parser.column_count()}")
    print(f"First row: {parser.get_row(0)}")
    print(f"Names: {parser.get_column(0)}")
    print(f"Ages: {parser.get_column(1)}")