在 Python 中将 .csv 值作为单个列表导入

Posted 2023-02-23

【中文标题】在 Python 中将 .csv 值作为单个列表导入

【英文标题】：Importing .csv values as single list in Python

【发布时间】：2018-02-25 06:25:08

【问题描述】：

我有一个 csv 文件，其中有一列记录如下所示：

test1
test2
test3
...

我使用以下方法将它们导入到 Python 列表中：

import csv

results = []
with open('test.csv', newline='') as inputfile:
    for row in csv.reader(inputfile):
        results.append(row)

print(results)

效果很好，但是，它会像这样打印出来：

[['test1'],['test2'],['test3']]

我将如何调整我的代码以将它们打印在一个列表中，而不是像这样：

['test1','test2','test3']

而且，一种输出是否比另一种更受欢迎？我意识到它依赖于用例。我只是好奇其他人如何通过 .csv 处理导入的列表，以及一个比另一个有什么优势。

谢谢！

【问题讨论】：

你没有 csv……你只有……文本……

确实如此。也许在这个特定的例子中，一个 .txt 文件会更合适。

【参考方案1】：

试试下面的代码

import csv

results = []
with open('test.csv', newline='') as inputfile:
    for row in csv.reader(inputfile):
        results.append(row[0])

print(results)

【讨论】：

这很好用。如果您的 csv 有多个列，但您只想将其中一列用作列表，只需更改 (row[0]) 中的索引号即可。