python - 使用 Python 创建 3D 数组

Question

我想在 Python (2.7) 中创建一个 3D 数组，以便像这样使用:

distance[i][j][k]

数组的大小应该是我拥有的变量的大小。 (nnn)

我尝试使用:

distance = [[[]*n]*n]

但这似乎不起作用。

我只能使用默认库，而乘法(即，[[0]*n]*n)不起作用，因为它们链接到同一个指针，我需要所有的值(value)观是个人的

Answer 1

您应该使用 list comprehension :

>>> import pprint
>>> n = 3
>>> distance = [[[0 for k in xrange(n)] for j in xrange(n)] for i in xrange(n)]
>>> pprint.pprint(distance)
[[[0, 0, 0], [0, 0, 0], [0, 0, 0]],
 [[0, 0, 0], [0, 0, 0], [0, 0, 0]],
 [[0, 0, 0], [0, 0, 0], [0, 0, 0]]]
>>> distance[0][1]
[0, 0, 0]
>>> distance[0][1][2]
0

您可以生成一个数据结构，其中包含一个看起来像您尝试过的语句的语句，但它会产生副作用，因为内部列表是按引用复制的:

>>> distance=[[[0]*n]*n]*n
>>> pprint.pprint(distance)
[[[0, 0, 0], [0, 0, 0], [0, 0, 0]],
 [[0, 0, 0], [0, 0, 0], [0, 0, 0]],
 [[0, 0, 0], [0, 0, 0], [0, 0, 0]]]
>>> distance[0][0][0] = 1
>>> pprint.pprint(distance)
[[[1, 0, 0], [1, 0, 0], [1, 0, 0]],
 [[1, 0, 0], [1, 0, 0], [1, 0, 0]],
 [[1, 0, 0], [1, 0, 0], [1, 0, 0]]]