2018.07.30
python 3.6
# 指定元素集
a = [1, 3]
data = {'A': (1,2,3), 'B': (4,5,6)}
df = pd.DataFrame(data, columns=['A', 'B'])
A B
0 1 4
1 2 5
2 3 6
# 得到新df_1: 删除'A'列中元素不在列表a中所在的行:
df_1 = df[df['A'].isin(a)]
A B
0 1 4
2 3 6
# 得到新df_2: 删除'A'列中元素不在列表a中所在的行:
df_2 = df[df['A'].isin(a).apply(lambda x: not x)]
A B
1 2 5
参考文献: