import csv
import numpy as np
import matplotlib.pyplot as plt
f=open('class.csv')
data=csv.reader(f)
h=next(data)
w=next(data)
h = np.array(h[2:], dtype = int)
w = np.array(w[2:], dtype = int)
for row in data :
    if '키' in row[1]:
        # concatenate(a, b) : 배열 a와 b를 병합하는 함수 # 3
        # 기존의 배열 h와 새로운 데이터를 병합하여 배열 h에 저장
        # axix는 배열의 차원을 결정하는 속성입니다. None : 1차원, 그 외: 다차원
        h = np.concatenate((h, np.array(row[2:], dtype = int)), axis = None)
    if '몸무게' in row[1] :
        # 기존의 배열 w와 새로운 데이터를 병합하여 배열 w에 저장
        w = np.concatenate((w, np.array(row[2:], dtype = int)), axis = None)
plt.scatter(h,w,c=w,cmap='jet')
plt.colorbar()
plt.show()