[Python/파이썬] numpy.average() 함수로 배열의 가중 평균 계산하기

numpy.average() 함수는 주어진 배열의 가중 평균을 계산하는 데 사용됩니다. 이 함수를 사용하여 배열의 각 요소에 대해 가중치를 적용하여 평균을 계산할 수 있습니다. 아래에서는 numpy.average() 함수를 사용하는 방법과 예제를 살펴보겠습니다.

numpy.average() 함수 예제

아래 예제에서는 numpy.average() 함수를 사용하여 주어진 배열의 가중 평균을 계산하는 방법을 보여줍니다.

import numpy as np

# 배열과 가중치 정의
arr = np.array([1, 2, 3, 4, 5])
weights = np.array([0.1, 0.2, 0.3, 0.2, 0.2])

# 가중 평균 계산
weighted_average = np.average(arr, weights=weights)
print("가중 평균:", weighted_average)

위의 코드는 주어진 배열 arr의 각 요소에 대해 가중치 weights를 적용하여 가중 평균을 계산합니다. numpy.average() 함수의 weights 매개변수를 사용하여 가중치를 전달합니다.

추가 예제

import numpy as np

# 다차원 배열의 가중 평균 계산
arr = np.array([[1, 2], [3, 4], [5, 6]])
weights = np.array([0.2, 0.8, 0.5])

# 가중 평균 계산
weighted_average = np.average(arr, axis=0, weights=weights)
print("가중 평균:", weighted_average)

위의 코드는 다차원 배열에 대한 가중 평균을 계산하는 예제입니다. numpy.average() 함수의 axis 매개변수를 사용하여 각 열에 대한 가중 평균을 계산합니다.