[파이썬 예제] Numpy 7- csv파일 읽고 합계 구하기

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

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

 

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

 

import csv
import numpy as np

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

header = next(lines)

a=[]

for line in lines:
    a.append(line)

row=(len(a))
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]

#csv 데이터중 6개열의 데이터만 리스트에 저장

data = np.insert(data, 6, 0, axis=1)
sum=0
for q in range(len(data)):
    sum=(data[q][1])+(data[q][3])+(data[q][5])
    data[q][6]+=sum
# data = np.insert(data, 6, 0, axis=1)
# row = data.shape[0]
# for i in range(row) :
#     sum = data[i][1] + data[i][3] + data[i][5]
#     data[i][6] = sum

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 ...   5 143 442]
 [  8 239   8 ...   8 270 737]
 [  6 182   6 ...   6 169 525]
 ...
 [  4  76   5 ...   5 103 270]
 [  4  99   4 ...   4  92 286]
 [  5 116   5 ...   5 126 340]]

 

 

 

 

 

 

 

 

 

 

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

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