Среднее значение столбца кадра данных python на основе другого столбца - PullRequest
0 голосов
/ 07 августа 2020

Я хотел бы взять среднее значение столбца b, когда соответствующее значение в столбце a> 5

Я получаю сообщение об ошибке:

TypeError: '>' не поддерживается между экземплярами из str и int

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
b = [0.05, 0.05, 0.05, 0.04, 0.03, 0, 0, 0, 0, 0.03]

d = {'col_a': a, 'col_b': b}
df = pd.DataFrame(d)

x = df['col_a' > 5]['col_b'].mean()
print(x)

1 Ответ

3 голосов
/ 07 августа 2020

df['col_a' > 5]

Это пытается проверить, соответствует ли строка 'col_a' > 5, что невозможно.

Вы имели в виду df[df['col_a'] > 5]['col_b'].mean()

...