numpyの多次元配列のリスト化

Question

numpyで多次元配列のリスト化みたいなことがしたいです。

arr_list = np.array([])
for k in range(3):
    data = randint(-20, 20, 80)
    # 20×４の行列
    arr = data.reshape(20, 4) 
    arr_list = np.stack([arr_list, arr])

以上のコードを実行すると、

ValueError                                Traceback (most recent call last)
<ipython-input-145-74f701246da9> in <module>()
----> 1 z_list = np.stack([z_list, z1])

~/.pyenv/versions/3.6.1/lib/python3.6/site-packages/numpy/core/shape_base.py in stack(arrays, axis, out)
    414     shapes = {arr.shape for arr in arrays}
    415     if len(shapes) != 1:
--> 416         raise ValueError('all input arrays must have the same shape')
    417 
    418     result_ndim = arrays[0].ndim + 1

ValueError: all input arrays must have the same shape

空の配列では、np.stackが使えませんでした。

20×4の行列をリスト化していき、arr_list.shape(3, 20, 4)のような形にしたいです。
arr_list[0]と指定すると、1番最初に作られた行列を抽出できるような仕組みを理想としています。

どのようにしたら実現できるでしょうか。

よろしくお願いいたします。

Answer 1

一度リストに入れてから結合するという流れを踏んでください。

import numpy as np
tmp = []
for k in range(3):
    data = np.random.randint(-20, 20, 80)
    arr = data.reshape(20, 4) 
    tmp.append(arr)

result = np.stack(tmp)
print(result.shape)  # => (3, 20, 4)