python - 从 3 维数组生成列表

Question

我在数组问题上苦苦挣扎了一段时间。例如，如果我有以下 3 维数组:

import os
import numpy as np

myArray = [[[1,2,3],
            [0,0,0],
            [2,2,2]],
           [[0,1,2],
            [1,1,1],
            [3,3,3]],
           [[2,3,4],
            [5,6,7],
            [8,9,10]]]

我想以递归方式为 myArray[0] 中的每个位置定义一个列表，该列表将存储 Z 轴上的值，以便:

list1 = [myArray[0][0][0],myArray[1][0][0], myArray[2][0][0]]  # would be [1,0,2]
list2 = [myArray[0][0][1],myArray[1][0][1], myArray[2][0][1]]  # would be [2,1,3]
list3 = [myArray[0][0][2],myArray[1][0][2], myArray[2][0][2]]  # would be [3,2,4]
list4 = [myArray[0][1][0],myArray[1][1][0], myArray[2][1][0]]  # would be [0,1,5]
list5 = [myArray[0][1][1],myArray[1][1][1], myArray[2][1][1]]  # would be [0,1,6]
list6 = [myArray[0][1][2],myArray[1][1][2], myArray[2][1][2]]  # would be [0,1,7]
list7 = [myArray[0][2][0],myArray[1][2][0], myArray[2][2][0]]  # would be [2,3,8]
list8 = [myArray[0][2][1],myArray[1][2][1], myArray[2][2][1]]  # would be [2,3,9]
list9 = [myArray[0][2][2],myArray[1][2][2], myArray[2][2][2]]  # would be [2,3,10]

我想它必须类似于

for i in range(len(myArray[0])-1):
    list+str(i) =[something]

你知道如何生成它吗？

非常感谢

Answer 1

由于您已经在使用 numpy:

np.swapaxes(np.swapaxes(myArray,0,2),0,1)

输出:

array([[[ 1,  0,  2],
        [ 2,  1,  3],
        [ 3,  2,  4]],

       [[ 0,  1,  5],
        [ 0,  1,  6],
        [ 0,  1,  7]],

       [[ 2,  3,  8],
        [ 2,  3,  9],
        [ 2,  3, 10]]])