python - 在这种情况下如何设置数组索引？

Question

我有这个文件

   0    0  716
   0    1  851
   0    2  900
   1    0  724
   1    1  857
   1    2  903
   2    0  812
   2    1  858
   2    2  902
   3    0  799
   3    1  852
   3    2  905
   4    0  833
   4    1  871
   4    2  907
   5    0  940
   5    1  955
   5    2  995
   6    0  941
   6    1  956
   6    2  996
   7    0  942
   7    1  957
   7    2  999
   8    0  944
   8    1  958
   8    2  992
   9    0  946
   9    1  952
   9    2  998

我想写这样的第三列值

0 0 716
1 0 724
2 0 812
3 0 799
4 0 833
0 1 851
1 1 857
2 1 858
3 1 852
4 1 871
0 2 900
1 2 903
2 2 902
3 2 905
4 2 907
5 0 940
6 0 941
7 0 942
8 0 944
9 0 946
5 1 955
6 1 956
7 1 957
8 1 958
9 1 952
5 2 995
6 2 996
7 2 999
8 2 992
9 2 998

我已读取文件

l= [line.rstrip('\n') for line in open('test.txt')]

现在我陷入困境，如何将其读取为 3d 数组？使用 enumerate 函数不起作用，因为它本身包含第一个值，我不需要它。

Answer 1

这有效:

with open('input.txt') as infile:
    rows = [map(int, line.split()) for line in infile]

def part(minval, maxval):
    return [r for r in rows if minval <= r[0] <= maxval]

with open('output.txt', 'w') as outfile:
    for half in [part(0, 4), part(5, 9)]:
        half.sort(key=lambda (a, b, c): (b, a, c))
        for row in half:
            outfile.write('%s %s %s\n' % tuple(row))

如果您有疑问，请告诉我。