为什么此代码会生成列表列表？答案

【问题标题】：Why Does This Code Produce a List of Lists?为什么此代码会生成列表列表？
【发布时间】：2018-06-22 03:35:09
【问题描述】：

我有一个函数：

def csvToList( filename ):
    with open(filename, 'r') as fp:
        reader = csv.reader(fp)
        myList = list(reader)
    return myList

并调用它：

fruitList = csvToList('fruit.csv')

fruit.csv的内容是：

apple,orange,kiwi,tomato

fruitList 的值是一个列表列表：

[['apple', 'orange', 'kiwi', 'tomato']]

为什么这段代码会生成一个列表列表，而不仅仅是一个简单的“平面”列表，如下所示：

['apple', 'orange', 'kiwi', 'tomato']

【问题讨论】：

【解决方案1】：

阅读器从 CSV 文件生成行。见documentation：

从 csv 文件中读取的每一行都作为字符串列表返回。

您的文件只包含 1 行并不重要； list(reader) 生成所有行的列表，可以是 0、1 或 2000 万行。因此，只有 1 行的文件会为您提供一个列表，其中包含该单行作为列表。

如果您只期望一行，请使用 next() function 迭代一步：

def csvToList( filename ):
    with open(filename, 'r') as fp:
        reader = csv.reader(fp)
        return next(reader, [])

next(reader, []) 告诉函数如果 reader 根本没有产生任何东西，则返回一个空列表。

【讨论】：