# 8 读取 Excel 文件|指定列(匹配)根据指定列名匹配读取让我们来个更难一点的,还是读取 某招聘网站数据.csv 文件,但现在有一个 list 中包含多个字段👇 # usecols = ['positionId','test','positionName', 'test1','salary']如果 usecols 中的列名存在于 某招聘网站数据.csv 中,则读取
时间: 2023-06-26 10:05:15 浏览: 329
读数据_读取文件某一列数据(带表头)_
5星 · 资源好评率100%
可以使用 pandas 库中的 read_csv 方法来读取 csv 文件,同时指定 usecols 参数来选择需要的列。可以使用 set() 函数将需要的列名放入一个集合中,然后遍历集合,判断列名是否存在于 csv 文件中,如果存在,则将该列读取出来。
以下是示例代码:
```python
import pandas as pd
# 读取 csv 文件
df = pd.read_csv('某招聘网站数据.csv', encoding='utf-8')
# 需要读取的列名列表
col_list = ['positionId', 'test', 'positionName', 'test1', 'salary']
# 列名集合
col_set = set(col_list)
# 读取数据
result = {}
for col in col_set:
if col in df.columns: # 判断是否存在该列
result[col] = df[col].tolist() # 读取该列数据
print(result)
```
输出结果为一个字典,字典中的键为需要读取的列名,值为该列的数据列表。
阅读全文