我有一個資料框:
id1 vals1 id2 vals2
a1 [5,6] b1 [8]
c1 [5,3] e1 [4,5,6]
我想計算 vals1 和 vals2 的平均值,如果這些平均值之間的差值的絕對值高于 1,則過濾我的資料框,以洗掉該行。怎么做?
所以想要的結果是:
id1 vals1 id2 vals2
c1 [5,3] e1 [4,5,6]
uj5u.com熱心網友回復:
你可以嘗試這樣的事情:
from statistics import mean
res = df.loc[df.apply(lambda x: mean(x.vals1) - mean(x.vals2),1).abs()<=1]
>>> res
'''
id1 vals1 id2 vals2
1 c1 [5, 3] e1 [4, 5, 6]
轉載請註明出處,本文鏈接:https://www.uj5u.com/ruanti/465172.html
標籤:Python python-3.x 数据框 功能
上一篇:如何根據熊貓中的條件洗掉一行?
