过滤并从列表生成子数据框

Question

我有一个set： CompanyList={'Apple','LG','Samsung'} 和pandasDataFrame：

sales=[{'name':'Samsung Korea','model':'S1'},
   {'name':'Samsung Vienam','model':'J1'},
   {'name':'LG America','model':'L1'}
  ]
df=pd.DataFrame(sales)

我想浏览CompanyList，然后从'sales'数据框生成新的子数据框。预期结果如下：

dataSamsung = [{'name': 'Samsung', 'model': 'S1'},{'name': 'Samsung', 'model': 'J1'}] 

dataLG = [{'name': 'LG', 'model': 'L1'}]

我试过：

 customer={}
 for i in companyList:
       customer[i] = df[df.name.str.contains('i')]

但这给了我一个错误的答案。你能帮我解决这个案子吗？你知道吗

Answer 1

试试apply：

df['name']=df['name'].apply(lambda x: [i for i in CompanyList if i in x][0])

apply与list comprehension。你知道吗