백준 알고리즘 4344번 평균은 넘겠지(python)

각 테스트마다 평균을 넘는 학생들의 비율을 출력해주면 된다. 

for _ in range(int(input())):
    li = list(map(int, input().split()))
    n = li[0]
    g_li = li[1:]
    average = sum(g_li)/n
    cnt = 0
    for g in g_li:
        if g > average:
            cnt += 1
    print('%.3f' %(cnt/n*100)+ '%')

아래는 numpy array를 사용한 코드이다. 위의 코드보다 성능이 좀 더 좋을 것 같다.

백준에서는 numpy가 import가 안되는지 런타임 에러가 나온다.

import numpy as np
for _ in range(int(input())):
    li = np.array(list(map(int, input().split())))
    ave = sum(li[1:])/li[0]
    print('%.3f' %(len(li[li > ave])/li[0]*100)+ '%')