pd.read_csv选择读入列数
时间: 2024-04-16 14:23:25 浏览: 82
使用pandas读取csv文件的指定列方法
在使用`pd.read_csv`函数读取CSV文件时,可以通过指定`usecols`参数来选择读入的列数。`usecols`参数可以接受一个列表或者一个函数作为输入。
如果传入一个列表,列表中的元素可以是列名或者列索引,表示只读取指定的列。例如,如果你只想读取CSV文件中的"列1"和"列3",可以这样设置`usecols`参数:
```python
import pandas as pd
data = pd.read_csv('data.csv', usecols=['列1', '列3'])
```
如果传入一个函数,函数会被应用到CSV文件的每一列上,返回值为True的列会被读取。例如,如果你只想读取列名以"列"开头的列,可以这样设置`usecols`参数:
```python
import pandas as pd
data = pd.read_csv('data.csv', usecols=lambda column: column.startswith('列'))
```
这样就只会读取以"列"开头的列。
阅读全文