我有一个数据框,其中包含产品ID,产品销售地点以及在该地区销售特定产品的团队。
我想为每个团队创建一个子数据框,如下所示:
team = ['red', 'blue', 'orange', 'black']
subdf_name = ['r', 'b', 'o' , 'bl']
我试过了:
def newdf (subdf, team_id):
subdf = df.loc [df ['Team'] == team_id , ['product_id', 'x', 'y', 'Team']]
return subdf
for i,j in zip (subdf_name, team_id) :
newdf (i,j)
然而,当我打电话给红队df(打印(红色),python报告
)NameError:未定义名称“red”
目前正在使用python 3.6
提前感谢您的合作