pd.read_csv取出指定列
时间: 2023-12-09 20:34:27 浏览: 84
pandas.read_csv参数详解(小结)
可以使用pd.read_csv()函数的usecols参数来取出指定列。usecols参数可以接受一个列表,列表中包含需要取出的列名或列号。例如,假设我们有一个名为data.csv的文件,其中包含三列数据,分别为A、B、C,我们想要取出B列和C列的数据,可以使用以下代码:
```python
import pandas as pd
# 读取文件,只取出B列和C列的数据
df = pd.read_csv('data.csv', usecols=['B', 'C'])
# 打印取出的数据
print(df)
```
如果我们想要取出第2列和第3列的数据,可以使用以下代码:
```python
import pandas as pd
# 读取文件,只取出第2列和第3列的数据
df = pd.read_csv('data.csv', usecols=[1, 2])
# 打印取出的数据
print(df)
```
阅读全文