dataframe 筛选所需行
import pandas as pd
data= pd.read_csv('path',names = ['user_id','item_id','rating','timestamp'])
user_cnt = data.groupby(['user_id']).size().to_frame('user_count').reset_index()
user_filter_id = user_cnt[user_cnt['user_count']>=10]['user_id'].tolist().
原创
2020-06-28 16:39:09 ·
1053 阅读 ·
0 评论