在 python 中编写代码以从“txt”文件中读取一些示例输入的最佳方法是什么？

Question

我知道这是一个非常基本的问题，但我也是 python 环境的新手。我正在编写我的第一个程序（数据结构问题），我需要阅读一些输入测试用例。

输入：

The first line contains the number of test cases T. T test cases follow. 
The first line for each case contains N, the number of elements to be sorted. 
The next line contains N integers a[1],a[2]...,a[N].

约束：

1 <= T <= 5
1 <= N <= 100000
1 <= a[i] <= 1000000

示例输入：

2
5
1 1 1 2 2
5
2 1 3 1 2

我编写了一个以下程序来从文件中读取上述输入，但我确信这不是最好的方法，因为它包含很多 if-else 循环和 for 循环，这真的很糟糕大号inputs。

sample = open('sample.txt')
first = sample.readline()
if len(first) > 5 or len(first) <1:
    print "Not correct input";
else:
    test = sample.readline
    for x in range(0,len(first)):
        second = sample.readline()
        if len(second) >100000 or len(second) < 1:
            print "wrong input";
        else:
            third = list()
            for y in range(0, len(third)):
                third.append(sample.readline()[:1])
        method_test(third)  #calling a method for each sample input

请建议我最好的解决方案。

Answer 1

应该这样做：

with open('sample.txt') as sample:
    num_testcases = int(sample.readline())
    assert 1 <= num_testcases <= 5
    for testcase in range(num_testcases):
        num_elems = int(sample.readline())
        assert 1 <= num_elems <= 10000
        elems = map(int, sample.readline().split())
        assert len(elems) == num_elems
        assert all(1 <= elem <= 100000 for elem in elems)
        method_test(elems)

编辑：添加了有效性检查。

Answer 2

首先。 len(x) 会告诉你输入的长度，所以如果你的输入行是“9”，len(line) 将是 1；如果您的输入行是“999”，len(line) 将是 3。您需要使用 int(line) 从输入文件中正确读取数字。

程序其余部分的逻辑看起来不正确-例如，您正在读取第一行（测试数），然后循环遍历该数字（这很好）-但您正在读取数字在这个循环之外的值，这是错误的顺序。

我强烈建议您在阅读时打印出各种值，这样您就可以了解正在发生的事情，并更轻松地调试您的程序。

最后当你执行以下操作时：

        third = list()
        for y in range(0, len(third)):...

您正在创建一个空列表list()，然后从 0 循环到列表的长度（也为零）。所以循环实际上不会做任何事情。

Answer 3

类似这样的：

使用cycle()只读取第一行之后的交替行，循环的大小将是T值的两倍。

 from itertools import islice,cycle
 with open("data1.txt") as f:
    T = int(f.readline())
    if T != 0:
            cyc=islice(cycle((False,True)),T*2) 
            for x in cyc:
                if x or not f.readline():
                    print map(int,f.readline().split())

输出：

[1, 1, 1, 2, 2]
[2, 1, 3, 1, 2]