從列表中篩選並生成子DataFrame

Question

我有一set ： CompanyList={'Apple','LG','Samsung'}和pandas DataFrame ：

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

我想瀏覽CompanyList，然后從“銷售” DataFrame生成新的Sub-DataFrame。 預期結果是

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 。