Python기초 8일차 <Module, Numpy>

실습 코드

# 각 사용자 별 평균 평점 구하기

user_mean_list = []
# for문을 사용해서 unique 데이터(user_id)에 접근
for number in user_id :
    # 1번 사용자부터 6040번 사용자까지 2번 인덱스 열의 값만 접근해서 평균 구함
    user_mean = data[data[:, 0] == number, 2].mean() # [조건식, 출력시킬 열]
    
    user_mean_list.append([number, user_mean])
    
    
user_mean_list

내가 쓴 코드

# 각 사용자별 평균 평점 구하기
for i in range(0,6040):
    rating = [user_id[i],np.mean(data[data[:,0] == i+1][:,2])]
    print(rating)

# (저장할때 파일명, 실제리스트명, 구분자, 데이터표현방식(소수점 3자리까지 표현))
np.savetxt("user_mean_list.csv", user_mean_list, delimiter="," , fmt="%.3f")