构建 numpy 数组的 pythonic 方式，无需显式处理第一个赋值

Question

我发现自己经常遇到这种编码模式：

matrices = np.random.rand(10, 10, 10)
block = np.array([])
for idx, M in enumerate(matrices):
    print(M.shape)
    if idx == 0:
        block = M
    else:
        block = np.dstack((block, M))

这只是一个例子，我知道在这种情况下我可以做block = matrices.有没有更好的方法来做到这一点，而不必处理

if idx == 0:
    block = M

箱？

Answer 1

不要用dstack(或任何连接变体(累加数组 - 使用列表附加

block = []
for idx, M in enumerate(matrices):
    block.append(M)

然后"堆叠"它们之后

block = np.dstack(block)   # concatenate(block, axis=2)

这更快，并避免了这种初始化尴尬。