首页 > 解决方案 > 如何根据案例数量分隔输入文件?

问题描述

我在 Python 中有一个编程问题,我根据输入文件回答。在这个测试文件中:

2
5
0.0
25.0
50.0
75.0
100.0
6
12.3
-67.1
122.8
428.4
-15.9
221.0

第一行“2”表示此文件中存在的案例数。我们还有 '5' 和 '6',它们代表案例中的浮点数。如何根据案例数量拆分文件?

标签: pythonfile

解决方案


考虑到输入在一个名为 的文件中input.txt,您可以使用类似这样的东西单独读取浮点数

with open('input.txt', 'r') as f:
no_of_test_cases = int(f.readline().strip())
i = 0
while i < no_of_test_cases:
    no_of_floats = int(f.readline().strip())
    floats = []
    for _ in range(no_of_floats):
        floats.append(float(f.readline().strip()))
    print(floats)
    # Do your calculations/operations
    i += 1

这使

[0.0, 25.0, 50.0, 75.0, 100.0]
[12.3, -67.1, 122.8, 428.4, -15.9, 221.0]

推荐阅读