pandas - 获取列中具有多个特定值的所有客户

Question

我该怎么做？:

获取所有提供了名字 (name_fist) 和生日的客户。

df = pd.DataFrame({
     'customer_ID': ['a','a', 'a', 'b', 'c', 'c', 'c', 'd', 'd'],
     'parameter': ['name_fist', 'name_last', 'birthday', 'name_fist', 
                   'name_fist', 'name_last', 'birthday', 'name_fist', 'name_last']})

customer_ID   parameter
0   a    name_fist
1   a    name_last
2   a    birthday
3   b    name_fist
4   c    name_fist
5   c    name_last
6   c    birthday
7   d    name_fist
8   d    name_last

答案应该是“2”。

理想情况下，我会获得满足此条件的所有人员的客户 ID:a 和 c

Answer 1

你可以尝试这样的事情:

(df.query('parameter in ["name_fist", "birthday"]')\
   .groupby('customer_ID')['parameter'].nunique() == 2)

输出:

customer_ID
a     True
b    False
c     True
d    False
Name: parameter, dtype: bool

详细信息:

首先使用query过滤数据帧以获取感兴趣的参数，然后groupby并使用nunique对参数进行计数。