有没有更好的pythonic方式来编写此代码?

Tom*_*der 1 python

我正在尝试从文本文件中读取第一列和第三列,并将它们添加在一起。

下面的代码可以完美地工作,并为我提供所需的结果,但是尝试找出是否有更好的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)
Run Code Online (Sandbox Code Playgroud)

随机文件实际上是一个随机文件。

col1,col2,col3
44,65,78
55,87,98
12,32,62
Run Code Online (Sandbox Code Playgroud)

结果:

[122, 153, 74]
Run Code Online (Sandbox Code Playgroud)

AGN*_*zer 6

如果可以使用,numpy那么我的建议是使用loadtxt功能:

import numpy as np
np.loadtxt('random.txt', dtype=int, skiprows=1, delimiter=',', usecols=(0, 2)).sum(axis=1).tolist()
Run Code Online (Sandbox Code Playgroud)