Question

我有2个不同的txt文件中的数据，我需要来自两者的数据（数据文件只有630行，但我宁愿找到一种方法来读取相同的行，而不仅仅是将相应的列复制并粘贴到一个文件中）。这两个文件的行数相同，顺序相同

这是我目前正在尝试使用的代码：

import numpy as np
class Source:
    def __init__(self, ra, dec, dist, knAPPMag, SGRB, incl, fif_area):
        self.ra = ra
        self.dec = dec
        self.dist = dist
        self.knAPPMag = knAPPMag
        self.SGRB = SGRB
        self.incl = incl
        self.fif_area = fif_area
def read_dat(data, data2):
    sources = {}
    count = 0
    for line in data:
        if line.startswith('#'):
            continue
        cols = line.split()
        ra = float(cols[3])
        dec = float(cols[4])
        dist = (float(cols[8]))*10e6
        incl = float(cols[5])
        for line2 in data2:
            if line2.startswith('#'):
                continue
            if line == line2:
                cols2 = line2.split()
                fif_area = float(cols2[8])

                knABMag = np.random.uniform(-13, -16)
                knAPPMag = 5*((np.log10(dist)) - 1 ) + knABMag

                SGRB_iso = np.random.uniform(1e+49, 1e+53) 
                dist_m = dist*3.08567758e+16 
                SGRB = SGRB_iso/(dist_m**2)

                name = 'src%05d' % count
                sources[name] = Source(ra, dec, dist, knAPPMag, SGRB, incl, fif_area)
                count=count+1



    return (sources)

我想创建一个包含两者条目的字典，但是当我尝试时，我得到一个空字典（上面的代码）我尝试过的其他事情是： 1）简单地在line循环之外运行line2循环，这留下630个几乎空的字典，2）取出line2 == line循环我返回396900个字典条目

我猜这个问题的解决方案并不复杂，我对python来说还不太新鲜干杯，杰克

Answer 1

我可能会做类似的事情：

from itertools import izip_longest as izipl

def read_pair(fna, fnb):
    lines = {}
    with open(fna, 'r') as fa:
        with open(fnb, 'r') as fb:
            for (i,(la,lb)) in enumerate(izipl(fa, fb)):
                lines[i] = [la, lb]
    return lines

lines = read_pair('file_a.txt', 'file_b.txt')

lines将是一个字典，其整数索引对应于行号。字典的值将是一个2元素列表，第一个元素是来自file_a.txt的行，第二个元素是来自file_b.txt的行。

从两个不同的文本文件中读取相同的行

1 个答案: