python - 错误 : The truth value of a Series is ambiguous

Question

我正在尝试使用以下代码生成新列

list = ['LHR','-1','-3','LGW','MAD','SIN','KUL','JFK','HKG','PVG','IST','SDA','GLA']
for i in list:
    if plotdata.loc[plotdata['LOCATION'] == i] :
        plotdata['city'] = plotdata['LOCATION']
    else:
        plotdata['city'] = 'others'

我收到以下错误:

ValueError: The truth value of a DataFrame is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

数据类型是类别，请问为什么会出现这个错误？

Answer 1

不确定为什么会收到此错误。但是，最好不要循环遍历 pandas。

您可以通过将“城市”设置为“位置”的任何位置，然后在不在列表中时恢复为“其他”来避免该错误。

city_list = ['LHR','-1','-3','LGW','MAD','SIN','KUL','JFK','HKG','PVG','IST','SDA','GLA']
plotdata['city'] = plotdata['LOCATION']
plotdata.loc[~plotdata['city'].isin(city_list), 'city'] = 'other'