2

我正在学习 python，但我的脚本还存在一些问题。

我有一个类似的文件：

1 5
2 5
3 5
4 2
5 1
6 7
7 7
8 8

我想在连续的行中打印成对的数字 2-1，只需在第 2 列中找到它们，然后在第 1 列和第 2 列中打印结果。结果将与此类似：

4 2 
5 1

我正在尝试用 python 来做，因为我的文件有 4,000,000 个数据。所以，这是我的脚本：

import linecache

final_lines = []
with open("file.dat") as f:
for i, line in enumerate(f, 1):
    if "1" in line:
        if "2" in linecache.getline("file.dat", i-1):
            linestart = i - 1 
            final_lines.append(linecache.getline("file.dat", linestart))
print(final_lines)

结果是：

['2\n', '2\n', '2\n']

我必须在我的脚本中更改什么以适应我想要的结果？，你能指导我吗？非常感谢。

4

3 回答 3

1

使用enumerate带有if语句的 for 循环来条件行，然后如果条件为真，则将两行附加到列表中final_lines：

final_lines = []
with open('file.dat') as f:
    lines = f.readlines()
    for i,line in enumerate(lines):
        if line.split()[1] == '2' and lines[i+1].split()[1] == '1':
            final_lines.extend([line,lines[i+1]])

现在：

print(final_lines)

将返回您想要的列表。

于 2019-03-22T01:58:35.733 回答
0

我认为会起作用

import re
with open("info.dat") as f:
   for match in re.findall("\d+ 2[\s\n]*\d+ 1",f.read()):
       print match

另见：https ://repl.it/repls/TatteredViciousResources

另一种选择是

lines = f.readlines()
for line,nextline in zip(lines,lines[1:]):
    if line.strip().endswith("2") and nextline.strip().endswith("1"):
       print(line+nextline)
于 2019-03-22T01:59:08.500 回答
0

你是 Python 的初学者，这很棒，所以我将采用更初级的方法。这是一个巨大的文件，因此您最好一次读取一行并仅保留该行，但实际上您需要两行来识别模式，因此请保留两行。考虑以下：

    fp = open('file.dat')
    last_line = fp.readline()
    next_line = fp.readline()
    while next_line:
        # logic to split the lines into a pair 
        # of numbers and check to see if the 
        # 2 and 1 end last_line and next_line
        # and outputting
        last_line = next_line
        next_line = fp.readline()

这遵循良好、可读的软件模式，并且需要最少的资源。

于 2019-03-24T05:41:08.700 回答