[파이썬 예제] Numpy 6- csv파일 읽고 리스트로 저장, 리스트를 ndarray에 저장

●CSV예제 파일을 다운받아 주세요

**입력하는 코드가 있는 파일과 같은 폴더에 위치해 있어야 합니다**

 

[파이썬 예제] Numpy 5- csv파일 읽고 리스트로 저장

 

import csv
import numpy as np

f = open('high_school_2019.csv', 'r', encoding='utf-8')
lines = csv.reader(f)

header = next(lines)
a=[]

# high_school_2019.csv의 내용을 한줄씩 읽어 append()를 사용하여 a리스트에 추가
for line in lines:
    a.append(line)

row=(len(a))

#a의 길이를 row에 저장
data=[]
data=np.zeros((len(a),6), dtype='int32')

for i in range(len(a)):
    for j in range(6):
        data[i][j] = a[i][j+2]

# 6개의 열(1학년 학급수, 1학년 학생수, 2학년 학급수, 2학년 학생수, 3학년 학급수, 3학년 학생수) 데이터를 

#ndarray에 저장

print('-'*60)
for i in range(6) :
    print(header[i+2], end=' ')

print()
print('-'*60)
print(data)

f.close()

 

------------------------------------------------------------
 1학년 학급수  1학년 학생수  2학년 학급수  2학년 학생수  3학년 학급수  3학년 학생수 
------------------------------------------------------------
[[  5 150   5 149   5 143]
 [  8 239   8 228   8 270]
 [  6 182   6 174   6 169]
 ...
 [  4  76   5  91   5 103]
 [  4  99   4  95   4  92]
 [  5 116   5  98   5 126]]

 

 

 

 

 

 

 

 

 

 

 

출처: 예제 중심 파이썬 입문

궁금한 부분이 있으면 댓글 부탁드립니다^^