프로그래밍/Python

[Python] csv파일 읽은 후 list생성하기

Victory_HA 2023. 9. 25. 10:24
  • python에서 csv파일을 읽은 후 list생성해봅시다csv 파일 내용csv 읽은 후 리스트 생성 및 특정 값을 출력해봅시다.

csv파일

label    class
0    100
1    101
2    102
3    103
4    104
0    200
1    201
2    202
3    203
4    204
5    205
6    206
7    207
8    208
9    209
10    210

data type 정의 후 2번째 값 출력

import csv

class DataEntry:
    def __init__(self, step_oh_name, step_oh, gt_name, gt_label, class_id):
        self.step_oh_name = step_oh_name
        self.step_oh = step_oh
        self.gt_name = gt_name
        self.gt_label = gt_label
        self.class_id = class_id

datas = []

csv_file_path = 'C:/Users/seungha.lee/Desktop/gts_class_map.csv'   

with open(csv_file_path, 'r', encoding='utf-8') as csv_file:
    csv_reader = csv.reader(csv_file)

    header = next(csv_reader)

    for row in csv_reader:
        data_entry = DataEntry(*row)
        datas.append(data_entry)

print(f"{datas[1].class_id}")