我想在巨大的数据框中找到特定的行,并在满足条件的情况下分配注释。
我使用的部分代码如下:
def check_car(x):
if x['CAR'] == 'FERRARI' and x['COUNTRY'] != 'ITALY':
return 'This is wrong'
else:
x.drop()
之后,我使用此功能创建带有ID和注释的新数据框。
新数据帧包含具有“This is wrong”
和None
值的ID。
而不是我想从数据帧中删除所有None值,是否可以在函数本身中做到这一点?
我使用了x.drop
,但出现了错误:
ValueError: ("Need to specify at least one of 'labels', 'index' or 'columns'”)
答案 0 :(得分:0)
首先选择满足您条件的所有列,然后为其分配值。 试试:
df = df.loc[ (df['CAR'] == 'FERRARI') & (df['COUNTRY'] != 'ITALY')]
df.loc["COLUMN" ] = "This is wrong"