Question

我在python中使用5x5x5x5x5的5维数组。

如何制作

的功能

将采用整数i（1 <= i <= 5）来迭代哪个维度，以及要锁定的其他4个维度[a，b，c，d]的列表
返回长度为5的一维列表

例如：

＆gt;＆gt;＆gt; bb =（5维数组）

＆gt;＆gt;＆gt; rowExtract（3，[2,3,0,2]，bb）

* [bb [2] [3] [i] [0] [2] for i in range（5）]返回*

3表示第3维，2,3,0,2表示其他维度

我可以通过硬编码5种不同的场景来实现，但有更简单的方法吗？

Answer 1

以下是使用循环的一种方法。

def extract_row(row, arr, bb):
  N = len(arr) + 1

  pre_mat = bb
  for i in range(row):
    pre_mat = pre_mat[arr[i]]

  ans = [] 
  for i in range(N):
    mat = pre_mat[i]
    for j in range(row + 1, N):
      mat = mat[arr[j - 1]]

    ans.append(mat)

  return ans

Answer 2

我认为没有别的办法。像这样的东西

def extractRow(row, arr, bb):
    if row == 0:
        return [bb[i][arr[0]][arr[1]][arr[2]][arr[3]] for i in range(5)]
    elif row == 1:
        return [bb[arr[0]][i][arr[1]][arr[2]][arr[3]] for i in range(5)]
    ...

Answer 3

一种可能的解决方案是使用迭代器来生成索引序列。＆＃34;失踪＆＃34;维度可以插入到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）

3 个答案: