python – 根据“not in”条件从数据帧中删除行

当日期列的值在日期列表中时,我想从pandas数据帧中删除行.以下代码不起作用：

a=['2015-01-01' , '2015-02-01']

df=df[df.datecolumn not in a]

我收到以下错误：

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

解决方法:

你可以使用pandas.Dataframe.isin.

pandas.Dateframe.isin将返回布尔值,具体取决于每个元素是否在列表a中.然后使用〜将其反转为将True转换为False,反之亦然.

import pandas as pd

a = ['2015-01-01' , '2015-02-01']

df = pd.DataFrame(data={'date':['2015-01-01' , '2015-02-01', '2015-03-01' , '2015-04-01', '2015-05-01' , '2015-06-01']})

print(df)
#         date
#0  2015-01-01
#1  2015-02-01
#2  2015-03-01
#3  2015-04-01
#4  2015-05-01
#5  2015-06-01

df = df[~df['date'].isin(a)]

print(df)
#         date
#2  2015-03-01
#3  2015-04-01
#4  2015-05-01
#5  2015-06-01