以下是一个示例代码,用于保留具有特定ID的所有记录:
import pandas as pd
# 创建示例数据
data = {'ID': [1, 2, 3, 4, 5],
'Name': ['Alice', 'Bob', 'Charlie', 'David', 'Eve']}
df = pd.DataFrame(data)
# 要保留的ID列表
ids_to_keep = [2, 4]
# 使用isin函数筛选数据
filtered_df = df[df['ID'].isin(ids_to_keep)]
# 输出筛选后的结果
print(filtered_df)
运行以上代码,将输出以下结果:
ID Name
1 2 Bob
3 4 David
通过使用isin
函数,我们可以将列"ID"中的每个值与要保留的ID列表进行比较。如果某个值在列表中,isin
函数将返回True
,否则返回False
。通过将这个布尔数组传递给DataFrame的索引,我们可以保留具有特定ID的所有记录。
上一篇:保留特定的标签