• 5조 문제

import numpy as np

# 사용자 수 입력
n = int(input("총 사용자 수를 입력하세요: "))

# 키, 몸무게, 성별 랜덤 생성
height = np.random.randint(160, 201, size = n )
weight = np.random.randint(50, 101, size = n )
sex = np.random.randint(0, 2, size = n )
arr = np.array([height, weight, sex]).T
print(arr)

#평균
male_h = arr[arr[:,2] == 0][:,0]
male_w = arr[arr[:,2] == 0][:,1]
female_h = arr[arr[:,2] == 1][:,0]
female_w = arr[arr[:,2] == 1][:,1]
print("남성 평균 신장:", round(np.mean(male_h),2))
print("남성 평균 몸무게", round(np.mean(male_w),2))
print("여성 평균 신장:", round(np.mean(female_h),2))
print("여성 평균 몸무게", round(np.mean(female_w),2))

#?????;;;