[파이썬 예제] Numpy 8- 각 학년별 총 학생수 구하기, 막대 그래프 시각화

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

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

 

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

 

 

import csv
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('font', family='Malgun Gothic')

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]

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_sum = np.sum(data, axis=0) # 총 학생수를 나타낼 ndarray 객체입니다.

print('1학년 총 학생수 :', data_sum[1])
print('2학년 총 학생수 :', data_sum[3])
print('3학년 총 학생수 :', data_sum[5])

xdata = ['1학년', '2학년', '3학년']

plt.bar(xdata, [data_sum[1], data_sum[3], data_sum[5]])
plt.title('전국 고등학교 학년별 총 학생수')
plt.savefig("plot0.png")
    

f.close()

 

1학년 총 학생수 : 463400
2학년 총 학생수 : 444714
3학년 총 학생수 : 500698

 

 

 

 

 

 

 

 

 

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

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