假设我有一个数据框:
first_df = pd.DataFrame({"company" : ['abc','def','xyz','lmn','def','xyz'],
"art_type": ['300x240','100x600','400x600','300x240','100x600','400x600'],
"metrics" : ['imp','rev','cpm','imp','rev','cpm'],
"value": [1234,23,0.5,1234,23,0.5]})
first_df = first_df.append(first_df)
我想要删除所有在列表['lmn','xyz']中具有公司值的行,并将其存储在另一个数据框中。
company_list = ['lmn', 'xyz']
我尝试了这个:
deleted_data = first_df[first_df['company'] in company_list]
这显然行不通,因为它是列表中的列表。使用for循环是解决这个问题的方法,还是有更好的方法?
for循环代码:
deleted_data = pd.DataFrame()
for x in company_list:
deleted_data = deleted_data.append(first_df[first_df['company']==x])