我有一个熊猫dataframe df
import pandas
df = pandas.DataFrame(
data=[["A", "Man"], ["A", "Woman"], ["A", "Man"], ["A", "Man"], ["B", "Woman"]],
columns=["category", "gender"],
)
df
category gender
0 A Man
1 A Woman
2 A Man
3 A Man
4 B Woman
我数了一下每个类别里有多少男人和女人
grouped = df.groupby(by=["category", "gender"])["gender"].count()
grouped
category gender
A Man 3
Woman 1
B Woman 1
Name: gender, dtype: int64
我怎样才能得到一个男人和女人都多于1的类别列表?
category_list = [A]
问题来源StackOverflow 地址:/questions/59382662/use-conditions-after-groupby
版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。
IIUC,
s = df.groupby('category')['gender'].value_counts().unstack(fill_value=0)
s[s.ge(1).all(1)]
给你
gender Man Woman
category
A 3 1