python读取csv文件中某几列代码
时间: 2023-12-01 15:41:52 浏览: 166
Python读取csv文件数据
以下是两种Python读取CSV文件中某几列的代码:
1. 使用csv模块读取CSV文件中的某几列:
```python
import csv
# 打开CSV文件
with open('data.csv', 'r') as f:
# 创建CSV阅读器
reader = csv.reader(f)
# 读取每行数据
for row in reader:
# 提取第一列和第三列数据
col1 = row[0]
col3 = row[2]
# 处理数据...
```
注意:上面的代码假设CSV文件中的每行数据都是用逗号分隔的。如果CSV文件使用其他字符来分隔数据,那么可以使用csv.reader()的delimiter参数来指定分隔符。例如,如果使用分号来分隔数据,可以这样创建CSV阅读器:
```python
reader = csv.reader(f, delimiter=';')
```
2. 使用pandas库读取CSV文件中的某几列:
```python
import pandas as pd
# 读取CSV文件中的某几列
df = pd.read_csv('data.csv', usecols=['col1', 'col3'])
# 处理数据...
```
注意:上面的代码假设CSV文件中的列名分别为col1、col2、col3等。如果CSV文件中没有列名,可以使用header=None参数来指定。
阅读全文