python - 熊猫根据条件创建列

Question

所以我想要做的是添加一个新列，称为'Top Tier'MAKE'BMW, FORD, BENZ, CHEVROLET'将具有值'Top'的位置，而其他 MAKE 将具有'Not Top'

df['Top Tier'] = np.where(df['MAKE']==['FORD', 'BMW', 'BENZ' , 'CHEVROLET], 'Top', 'Not Top')

^这不起作用。我想使用 np.

标签： pythonpandas

Answer 1

你可以这样做：

df['Top Tier'] = df['MAKE'].map(lambda x: 'Top' if x in ['FORD','BMW','BENZ','CHEVROLET'] else 'Not Top')

或这个：

df['Top Tier'] = df['MAKE'].isin(['FORD','BMW','BENZ','CHEVROLET']).replace({True: 'Top', False: 'Not Top'})