pd.read_csv时做列筛选
时间: 2023-07-06 08:34:37 浏览: 86
使用pandas库对csv文件进行筛选保存
5星 · 资源好评率100%
在使用 `pd.read_csv` 读取文件时,可以使用 `usecols` 参数指定需要读取的列。该参数接受一个列表,列表中的元素为需要读取的列名或列索引。例如:
```python
import pandas as pd
df = pd.read_csv('data.csv', usecols=['col1', 'col3']) # 读取 col1 和 col3 两列数据
```
或者使用列索引:
```python
df = pd.read_csv('data.csv', usecols=[0, 2]) # 读取第1列和第3列数据
```
注意:如果文件中的列名与指定的列名不匹配,将会抛出一个 `ValueError` 异常。
阅读全文