有没有更好的pythonic方法来编写这个代码？

Question

我试图从文本文件中读取第一列和第三列，并将它们添加到一起。你知道吗

下面的代码工作得很完美，并给出了我需要的结果，但是否有更好的更pythonic的方法来写这个呢？你知道吗

with open('random.txt', 'r') as fn:
    next(fn)
    numbers = fn.readlines()
    first_col = [int(x.split(',')[0]) for x in numbers]
    third_col = [int(y.split(',')[2]) for y in numbers]

    result = [v + z for v, z in zip(first_col, third_col)]

    print(result)

随机文件实际上是一个随机文件。你知道吗

col1,col2,col3
44,65,78
55,87,98
12,32,62

结果：

[122, 153, 74]

Answer 1

您可以使用zip：

with open('random.txt', 'r') as fn:
    next(fn)
    first_col, _, third_col  = [
        *zip(*(int(x) for x in map(lambda x: x.split(','), fn))
    ]
    ...
    results = [x+y for x, y in zip(first_col, second_col)]

或者如果你不需要抱着驴：

results = [
    x+y for x, _, y in zip(*(int(x) for x in map(lambda x: x.split(','), fn))
]