How to filter dataset by difference of mean values in columns?

user_17378883 0
Viewed 58

I have a dataframe:

id1     vals1   id2    vals2
a1      [5,6]   b1     [8]
c1      [5,3]   e1     [4,5,6]

I want to calculate mean values of vals1 and vals2 and filter my dataframe if absolute value of difference between those mean values is higher than 1, to remove that row. How to do that?

So desired result is:

id1     vals1   id2    vals2
c1      [5,3]   e1     [4,5,6]
dataframe python-3.x function python
1 Answers

you can try something like this:

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]
user_183445120
Related