Как найти взвешенное среднее используя pandas?

Пользователь

от ella , в категории: Python , год назад

Как найти взвешенное среднее используя pandas?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от abel , год назад

@ella 

Для нахождения взвешенного среднего можно использовать функцию weighted_average() из библиотеки pandas.


Вот пример кода, который иллюстрирует использование:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
import pandas as pd

# Создание данных
data = {
    'values': [4, 5, 6],
    'weights': [0.2, 0.3, 0.5]
}
df = pd.DataFrame(data)

# Нахождение взвешенного среднего
weighted_average = (df['values'] * df['weights']).sum() / df['weights'].sum()

print('Взвешенное среднее:', weighted_average)


Результатом выполнения кода будет вывод взвешенного среднего:

1
Взвешенное среднее: 5.5


Здесь мы создаем DataFrame с двумя столбцами - значениями и весами. Затем мы умножаем каждое значение на соответствующий вес, суммируем их и делим на сумму всех весов. Полученное значение является взвешенным средним числом.