dateframe中保留某列等于一某个list
时间: 2023-08-06 09:08:42 浏览: 80
Python将DataFrame的某一列作为index的方法
您可以使用DataFrame的isin()方法来筛选出某列等于给定列表中的值的行。以下是示例代码:
```python
import pandas as pd
# 创建示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
'B': [6, 7, 8, 9, 10],
'C': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)
# 给定列表
values_to_keep = [2, 4]
# 保留某列等于给定列表中的值的行
filtered_df = df[df['A'].isin(values_to_keep)]
print(filtered_df)
```
输出结果为:
```
A B C
1 2 7 12
3 4 9 14
```
在上述示例中,我们创建了一个DataFrame,并指定了一个名为'A'的列。然后,我们定义了一个列表`values_to_keep`,其中包含我们希望保留的值。最后,我们使用`isin()`方法筛选出'A'列等于给定列表中的值的行,并将结果存储在`filtered_df`中。
阅读全文