python pandas groupby filter

Question

Concretely, say I have a DataFrame like this:

appid  mac_id  count

1      a        1 

2      b        1

2      c        1

3      d        1

3      e        1

And I also have a :

mac_list = ['b', 'd', 'e']

I want to group this data frame on appid and for every group filter mac_id if it's in mac_list. Last, sum(count) for every group.

for this DataFrame the result is:

appid   count

1         0

2         1

3         2

How can I do this with Pandas?

Katriel · Accepted Answer · 2013-07-12 07:34:12Z

7

>>> df = pd.DataFrame({"appid": [1,2,2,3,3], "mac_id": ['a', 'b', 'c', 'd', 'e'], "count": [1,1,1,1,1]})
>>> summer = lambda x: x[x["mac_id"].isin(mac_list)].sum()
>>> df.groupby("appid").apply(summer)["count"]
  18 
appid
1        0
2        1
3        2
Name: count, dtype: object

answered Jul 12, 2013 at 7:34

Katriel

124k19 gold badges141 silver badges172 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

python pandas groupby filter

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related