我需要制作一个多维零数组。
对于两个 (D=2) 或三个 (D=3) 维度,这很容易,我会使用:
a = numpy.zeros(shape=(n,n))
或者
a = numpy.zeros(shape=(n,n,n))
对于更高的 D,我如何制作长度为 n 的数组?
我需要制作一个多维零数组。
对于两个 (D=2) 或三个 (D=3) 维度,这很容易,我会使用:
a = numpy.zeros(shape=(n,n))
或者
a = numpy.zeros(shape=(n,n,n))
对于更高的 D,我如何制作长度为 n 的数组?
您可以将元组乘以(n,)
所需的维数。例如:
>>> import numpy as np
>>> N=2
>>> np.zeros((N,)*1)
array([ 0., 0.])
>>> np.zeros((N,)*2)
array([[ 0., 0.],
[ 0., 0.]])
>>> np.zeros((N,)*3)
array([[[ 0., 0.],
[ 0., 0.]],
[[ 0., 0.],
[ 0., 0.]]])
>>> sh = (10, 10, 10, 10)
>>> z1 = zeros(10000).reshape(*sh)
>>> z1.shape
(10, 10, 10, 10)
编辑:虽然上面没有错,但这只是过度。@mgilson 的答案更好。
In [4]: import numpy
In [5]: n = 2
In [6]: d = 4
In [7]: a = numpy.zeros(shape=[n]*d)
In [8]: a
Out[8]:
array([[[[ 0., 0.],
[ 0., 0.]],
[[ 0., 0.],
[ 0., 0.]]],
[[[ 0., 0.],
[ 0., 0.]],
[[ 0., 0.],
[ 0., 0.]]]])
您可以使用方括号制作多维零数组
array_4D = np.zeros([3,3,3,3])