您可以将多维数组的位置存储在变量中吗？

Question

我正在尝试编写一个 rubix 立方体求解器，我想重复调用相同的边来检查它的信息。但是三维区域的位置占用了很多空间，我想把我重复使用的位置放到一个变量中，这样我就可以调用变量而不是一遍又一遍地重写位置。

这是我的 rubix 立方体的样子：

rubixCube = [
        [["G","G","Y"], ["R","B","O"], ["R","R","O"]],
        [["G","O","O"], ["O","Y","G"], ["R","B","B"]],
        [["Y","W","Y"], ["R","O","O"], ["O","W","W"]],
        [["B","B","W"], ["B","W","G"], ["G","W","R"]],
        [["B","Y","R"], ["Y","R","W"], ["G","Y","B"]], 
        [["O","G","W"], ["B","G","R"], ["Y","Y","W"]]
        ]

这是我反复调用的位置示例：

if(rubixCube[0][0][1] == "W"):

我可以写一些类似这样的东西吗：

position = [0][0][1]
if(rubixCube[position] == "W"):

Answer 1

如果您使用 numpy 数组

import numpy as np

rubixCube = np.array([
        [["G","G","Y"], ["R","B","O"], ["R","R","O"]],
        [["G","O","O"], ["O","Y","G"], ["R","B","B"]],
        [["Y","W","Y"], ["R","O","O"], ["O","W","W"]],
        [["B","B","W"], ["B","W","G"], ["G","W","R"]],
        [["B","Y","R"], ["Y","R","W"], ["G","Y","B"]],
        [["O","G","W"], ["B","G","R"], ["Y","Y","W"]]
        ])

然后你可以用元组索引它

pos = (0, 0, 1)
print(rubixCude[pos])