Python列表操作(追加多个列表)

Question

我有一个名为 test.txt 的文本文件，其中包含这种格式的数据

....
a|b|c|d|e
a1|b2|c3|d4|e5
a3|b5|c2|d1|e3
....

我想将每列的值放入列表中:像这样

list1=[a,a1,a3]
list2=[b,b2,b5]

我成功地通过这样做完成了这项工作:

list1,list2,list3,list4,list5 = ([] for i in range(5))

for line in open('test.txt','r'):
    temp=line.split('|')
    list1.append(temp[0])
    list2.append(temp[1])
    list3.append(temp[2])
    list4.append(temp[3])
    list5.append(temp[4].strip())

是否有更短的方法将值 append 到每个列表？我只能想到为每个列表使用 1 行，如上所述。

Answer 1

zip() 是你的 friend :

list1, list2, list3, list4, list5 = zip(
    *(line.strip().split('|') for line in open('test.txt')))

作为一个额外的好处，即使您不知道有多少列，您也可以使用它 - 只需将其分配给单个变量，您就会得到一个列表，其中的每一项都是值对于一列:

column_values = zip(*(line.strip().split('|') for line in open('test.txt')))
# column_values[0] is [a,a1,a3] ...

<小时/>

让我们逐步了解一下这个过程。首先，我们来看看仅使用 zip() 位会发生什么:

list1, list2, list3, list4, list5 = zip(
    [0,1,2,3,4], [0,1,2,3,4], [0,1,2,3,4])

结果为 list1 = [0,0,0] 等等，因为 zip() 从每个列表中获取第一个元素并将其放入列表中，如下所示结果的第一个元素。

<小时/>

现在，我们如何从序列[a,b,c]得到zip(a,b,c)？简单:我们使用 * 位置参数扩展运算符。 zip(*L) 与 zip(L[0], L[1], ...) 相同。

<小时/>

最后，我们如何获取我们需要传入的列表列表呢？我们使用生成器表达式:

(line.strip().split('|') for line in open('test.txt'))

创建一个生成器，生成每行中的项目列表，一次一行(并去除项目中的空格)。这正是我们需要提供给 zip() 以获得我们想要的结果的内容。