时间戳

首页 > 解决方案 > 如何从csv文件中读取前两列

python - 如何从csv文件中读取前两列

问题描述

我的任务是使用 python 读取 csv 文件。

任务是从 csv 文件中读取前两列。

csv 文件看起来像

car name,miles/gallon,cylinders,displacement,horsepower,weight,acceleration,model year,origin
"chevrolet chevelle malibu",18,8,307,130,3504,12,70,1
"buick skylark 320",15,8,350,165,3693,11.5,70,1
"plymouth satellite",18,8,318,150,3436,11,70,1

我的代码：

import csv
list2 = []
with open("mylist.csv") as f:
    for row in f:
        list2.append(row[0])

标签： pythoncsv

解决方案

pandas 具有指定要读取文件的哪些列的read_csv()内置功能。只需提供usecols一个整数值或列索引列表：

import pandas as pd
df = pd.read_csv('~/file.csv', usecols=[0,1], header=0)
print(df)

输出：

                    car name  miles/gallon
0  chevrolet chevelle malibu            18
1          buick skylark 320            15
2         plymouth satellite            18

推荐阅读