Фильтровать и генерировать подкадр из списка

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, а затем сгенерировать новый Sub-DataFrame из продаж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.