pandas判断一列数据是否在列表中
时间: 2023-05-22 08:01:52 浏览: 109
pandas对dataFrame中某一个列的数据进行处理的方法
5星 · 资源好评率100%
可以使用 isin() 方法来判断一列数据是否在列表中,示例代码如下:
```python
import pandas as pd
# 创建数据
df = pd.DataFrame({'A': ['foo', 'bar', 'baz', 'qux'],
'B': [1, 2, 3, 4]})
# 判断列 A 是否在列表 ['foo', 'baz'] 中
mask = df['A'].isin(['foo', 'baz'])
print(mask)
```
输出结果为:
```
0 True
1 False
2 True
3 False
Name: A, dtype: bool
```
其中,返回的 mask 是一个布尔型的 Series,表示每个元素是否在列表中。
阅读全文