基於另一列平均一個 python dataframe 列

Question

當a列中的相應值> 5時，我想取b列的平均值

我收到錯誤消息：

類型錯誤：“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)

Answer 1

df['col_a' > 5]

這試圖檢查字符串'col_a'是否> 5 ，這是無法完成的。

你的意思是df[df['col_a'] > 5]['col_b'].mean()