我有一个大约有2000条记录的CSV文件。
每条记录都有一个字符串和一个类别:
This is the first line,Line1
This is the second line,Line2
This is the third line,Line3
我需要把这个文件读入一个列表，看起来像这样:
data = [('This is the first line', 'Line1'),
('This is the second line', 'Line2'),
('This is the third line', 'Line3')]
如何使用Python将此CSV导入到我需要的列表?
当前回答
您可以使用list()函数将csv阅读器对象转换为列表
import csv
with open('input.csv', newline='') as csv_file:
reader = csv.reader(csv_file, delimiter=',')
rows = list(reader)
print(rows)
其他回答
您可以使用list()函数将csv阅读器对象转换为列表
import csv
with open('input.csv', newline='') as csv_file:
reader = csv.reader(csv_file, delimiter=',')
rows = list(reader)
print(rows)
如果确定输入中没有逗号，而不是分隔类别，则可以逐行读取文件并在，，上进行分割，然后将结果推入List
也就是说，看起来您正在查看一个CSV文件，因此您可以考虑为它使用模块
result = []
for line in text.splitlines():
result.append(tuple(line.split(",")))
接下来是一段代码，它使用csv模块，但使用第一行(csv表的头)将file.csv内容提取到dicts列表
import csv
def csv2dicts(filename):
with open(filename, 'rb') as f:
reader = csv.reader(f)
lines = list(reader)
if len(lines) < 2: return None
names = lines[0]
if len(names) < 1: return None
dicts = []
for values in lines[1:]:
if len(values) != len(names): return None
d = {}
for i,_ in enumerate(names):
d[names[i]] = values[i]
dicts.append(d)
return dicts
return None
if __name__ == '__main__':
your_list = csv2dicts('file.csv')
print your_list
Python3的更新:
import csv
from pprint import pprint
with open('text.csv', newline='') as file:
reader = csv.reader(file)
res = list(map(tuple, reader))
pprint(res)
输出:
[('This is the first line', ' Line1'),
('This is the second line', ' Line2'),
('This is the third line', ' Line3')]
如果csvfile是一个文件对象，它应该用newline= "打开。 csv模块
