如何动态创建一个三维数组

Question

如果我想要一个数组，例如:

[
    [
        [6,3,4],
        [5,2]
    ],
    [
        [8,5,7],
        [11,3]
    ]
]

我给你们一个简单的例子。实际上，每个维度的数组个数会随着条件的不同而改变。我不想用乘法运算。我想直接创建每个元素

怎么做?

谢谢!

Answer 1

使用多维索引到值的映射。不要用列表的列表的列表

array_3d = {
    (0,0,0): 6, (0,0,1): 3, (0,0,2): 4,
    (0,1,0): 5, (0,1,1): 2,
    (1,0,0): 8, (1,0,1): 5, (1,0,2): 7,
    (1,1,0): 11,(1,1,1): 3 
}

现在你不必担心"预分配"任何大小或维度数量或任何东西

Answer 2

对于这种情况，我一直使用字典:

def set_3dict(dict3,x,y,z,val):
  """Set values in a 3d dictionary"""
  if dict3.get(x) == None:
    dict3[x] = {y: {z: val}}
  elif dict3[x].get(y) == None:
    dict3[x][y] = {z: val}
  else:
    dict3[x][y][z] = val
d={}    
set_3dict(d,0,0,0,6)
set_3dict(d,0,0,1,3) 
set_3dict(d,0,0,2,4)
...

同样，我有一个getter

def get_3dict(dict3, x, y, z, preset=None):
  """Read values from 3d dictionary"""
  if dict3.get(x, preset) == preset:
    return preset
  elif dict3[x].get(y, preset) == preset:
    return preset
  elif dict3[x][y].get(z, preset) == preset:
    return preset
  else: return dict3[x][y].get(z)
>>> get3_dict(d,0,0,0)
 6
>>> d[0][0][0]
 6
>>> get3_dict(d,-1,-1,-1)
 None
>>> d[-1][-1][-1]
 KeyError: -1

在我看来，优点在于遍历字段非常简单:

for x in d.keys():
  for y in d[x].keys():
    for z in d[x][y].keys():
      print d[x][y][z]

Answer 3

嗯，和你想的差不多。在Python中，它们被称为列表，而不是数组，只是一个三层嵌套的列表，比如

threeDList = [[[]]]

然后使用三个索引来标识元素，比如

threeDList[0][0].append(1)
threeDList[0][0].append(2)
#threeDList == [[[1,2]]]
threeDList[0][0][1] = 3
#threeDList == [[[1,3]]]

你只需要注意你使用的每个索引都指向列表中已经存在的位置(例如，threeDList[0][0][2]或threeDList[0][1]或threeDList[1]在本例中不存在)，并且在可能的情况下，只使用推导式或for循环来操作列表中的元素。

希望这对你有帮助!