pandas读取csv文件某几列
时间: 2023-09-08 20:13:16 浏览: 126
CSV数据-Pandas-Tests-IO-Data-CSV-Tips-2024-01-09.zip
可以使用pandas库中的read_csv()函数来读取csv文件,并通过指定列名或列号来选择需要读取的列。下面是两种方法:
1. 通过列名选择:
```python
import pandas as pd
# 读取csv文件
df = pd.read_csv('data.csv')
# 选择需要读取的列名
cols = ['col1', 'col2', 'col3']
# 读取指定列
df_selected = df[cols]
```
2. 通过列号选择:
```python
import pandas as pd
# 读取csv文件
df = pd.read_csv('data.csv')
# 选择需要读取的列号
cols = [0, 2, 4]
# 读取指定列
df_selected = df.iloc[:, cols]
```
在第二种方法中,`iloc`函数中的`:`表示选择所有行,`cols`表示选择指定的列号。需要注意的是,列号从0开始计数。
阅读全文