题解 | 短视频用户数据和购买数据

import pandas as pd
#导入两个表，注意设置dtype
dt1=pd.read_csv('Tk.csv',dtype={'ID':int,'Follows':int,'Fans':int,'Cost':float,'Days':int,'Number_of_vedios':int})
dt2=pd.read_csv('Shop.csv',dtype={'ID':int,'order_num':str,'price':float,'date':str,'shop_name':str,'VIP_level':int})
#从表一中筛选出粉丝数大于10000的ID
a=dt1[dt1['Fans']>10000]['ID']
#建立一个空列表，存储dt2中匹配ID的索引
index_list=[]
for i in a:
    index_list.extend(dt2[dt2['ID']==i].index.tolist())
#从dt2检索索引列表，并且取shop_name列
s=dt2.loc[index_list,'shop_name'].sort_index()
print(s)