帮我优化一下这段代码user_mer = datasets[['User_id','Merchant_id']].groupby(by=['User_id','Merchant_id']).agg({'Merchant_id':'count'}) user_mer.columns = ['user_mer'] user_mer.reset_index(inplace=True)
时间: 2023-08-09 11:08:53 浏览: 81
DNA-k-mer-index.rar_DNA_DNA C#_K-Index_k-mer的意义_site:www.pudn.co
这段代码可以进行如下优化:
```python
user_mer = datasets.groupby(['User_id', 'Merchant_id']).size().reset_index(name='user_mer')
```
这里使用了 `groupby` 函数对 `User_id` 和 `Merchant_id` 进行分组,并使用 `size` 函数对每个分组的大小进行计数。然后使用 `reset_index` 函数将结果重新设置为 DataFrame,并给计数列起一个名字 `user_mer`。
相比原来的代码,这段代码更加简洁,同时也更加高效。
阅读全文