python - 如何从 CSV 文件中导入数据并将其存储在变量中？

Question

我是 python 3 的新手，我边走边学。我认为有人可以帮助我解决一个基本问题:如何将 CSV 文件中的文本存储为变量以供稍后在代码中使用。所以这里的想法是将 CSV 文件导入 python 解释器:

import csv
with open('some.csv', 'rb') as f:
    reader = csv.reader(f)
    for row in reader:
        ...

然后从该文件中提取文本并将其存储为变量(即 w = ["csv file text"])，以便稍后在代码中用于创建排列:

print (list(itertools.permutations(["w"], 2)))

如果有人可以帮助和解释这个过程，我将不胜感激，因为我真的很努力学习。如果需要更多解释，请告诉我!

Answer 1

itertools.permutations() 需要一个可迭代对象(例如列表)和一个长度作为其参数，因此您的数据结构需要反射(reflect)这一点，但您还需要定义您要尝试的内容在这里实现。例如，如果您想读取 CSV 文件并在每个单独的 CSV 字段上生成排列，您可以尝试这样做:

import csv
with open('some.csv', newline='') as f:
    reader = csv.reader(f)
    w = []
    for row in reader:
        w.extend(row)

print(list(itertools.permutations(w, 2)))

这里的关键是创建一个可以传递给 itertools.permutations() 的平面列表 - 这是通过将 w 初始化为一个空列表来完成的，并且然后使用 CSV 文件每一行的元素/字段扩展其元素。

注意:正如@martineau 所指出的，原因已解释 here ，当与 Python 3 csv 模块一起使用时，应使用 newline='' 打开文件。