gpt4 book ai didi

python - 根据列表填充数据框列

转载 作者:太空宇宙 更新时间:2023-11-04 00:19:02 27 4
gpt4 key购买 nike

我有一个数据框

vehicle_make    vehicle_model   vehicle_year
Toyota Corolla 2016
Hyundai Sonata 2016
Cadillac DTS 2006
Toyota Prius 2014
Kia Optima 2015

我想添加一个新列“vehicle_make_category”,它根据我拥有的列表进行填充

luxury=['Bentley',
'Maserati',
'Hummer',
'Porsche',
'Lexus']

non_luxury=['Saab',
'Mazda',
'Dodge',
'Volkswagen',
'Kia',
'Chevrolet',
'Hyundai',
'Ford',
'Nissan',
'Honda',
'Toyota'
]

如何做到这一点?我试过使用

df['vehicle_make_category']=np.where(df['vehicle_make']=i for i in luxury, 'luxury')

但它不起作用......

最佳答案

简单

df["vehicle_make_category"] = None
df.loc[df["vehicle_make"].isin(luxury), "vehicle_make_category"] = "luxury"
df.loc[df["vehicle_make"].isin(non_luxury), "vehicle_make_category"] = "non_luxury"

关于python - 根据列表填充数据框列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49864401/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com