我有一个数据集data,其中有一列customer_id,允许有重复的,有一列type,分为order,follow,pageview和comment。我想通过python代码查询同一个customer_id做出follow和order两种行为的客户,并展示出来,怎么写代码?
时间: 2024-04-30 09:22:00 浏览: 58
基于深度学习实现的声纹识别项目python源码+声纹数据集+项目说明文档.zip
5星 · 资源好评率100%
可以使用pandas库来解决这个问题。
首先,读取数据集并导入pandas库:
```python
import pandas as pd
data = pd.read_csv('data.csv')
```
然后,使用pandas的groupby和filter方法来查询同一个customer_id做出follow和order两种行为的客户。具体代码如下:
```python
result = data.groupby('customer_id').filter(lambda x: set(['follow', 'order']).issubset(set(x['type'])))
```
最后,输出结果:
```python
print(result)
```
这样就可以查询到同一个customer_id做出follow和order两种行为的客户了。
阅读全文