一个可能的解决方案使用迭代器来生成索引序列。 “缺失”维可以插入到arr
中,并且每次通过生成器循环时都会使用下一个值填充该索引。每次发生器产生时,它会生成一个5元素数组,其中包含从bb
中提取下一个值所需的索引。然后你可以从bb
中选择你想要的元素。
def extract_row(row, bb, arr):
retval = []
for a0, a1, a2, a3, a4 in iter_indices(row, arr):
retval.append(bb[a0][a1][a2][a3][a4])
return retval
def iter_indices(row, arr):
arr.insert(row, None)
for i in range(5):
arr[row] = i
yield arr
# TEST CODE FOLLOWS
g = 0
def deepen(b, n):
global g
if n == 1:
b.append(g)
b.append(g+1)
b.append(g+2)
b.append(g+3)
b.append(g+4)
g += 5
return b
for m in range(5):
b.append(deepen([], n-1))
return b
a = []
deepen(a, 5)
# Run it using the test matrix a
print(extract_row(1, a, [1, 2, 3, 4]))
print(a[1][0][2][3][4], a[1][1][2][3][4], a[1][2][2][3][4], a[1][3][2][3][4], a[1][4][2][3][4])
Output is:
[694, 819, 944, 1069, 1194]
694 819 944 1069 1194
嗯,我同意。我还没有在互联网上的其他地方看到过这样的问题。看起来Python似乎不支持这样一个特定的问题。谢谢! – marlont