python 判断列表和csv文件相同值并获取值
时间: 2023-06-05 07:07:29 浏览: 243
可以使用Python的pandas库来实现。首先读取csv文件,将其转换成DataFrame格式。然后将列表转换成Series格式,再使用isin()方法来判断两个数据结构中是否有相同的值。最后使用loc[]方法来获取相同的值。
代码示例:
```python
import pandas as pd
# 读取csv文件
df = pd.read_csv('data.csv')
# 列表
lst = [1, 2, 3, 4, 5]
# 转换成Series格式
s = pd.Series(lst)
# 判断是否有相同值
mask = s.isin(df['column_name'])
# 获取相同的值
result = s.loc[mask]
print(result)
```
其中,'column_name'为csv文件中需要比较的列名。
阅读全文