在python函数中从数据框中删除空行

时间:2019-06-12 11:06:12

标签: python pandas dataframe

我想在巨大的数据框中找到特定的行,并在满足条件的情况下分配注释。

我使用的部分代码如下:

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'”)

1 个答案:

答案 0 :(得分:0)

首先选择满足您条件的所有列,然后为其分配值。 试试:

df = df.loc[ (df['CAR'] == 'FERRARI') & (df['COUNTRY'] != 'ITALY')]
df.loc["COLUMN" ] = "This is wrong"