创建 np.array 填充零数组

Question

我正在尝试初始化一个“空”数组，每个元素都包含 t_list 一个 8x8 np.zeros 数组：

t_list = np.zeros((8,8), dtype=np.float32)

我现在想要一个 np.array 在每个索引处有多个 t_list：

result = np.array((t_list, t_list, ...., tlist))

我希望能够控制 t_list 在 result 中的次数。

我知道我可以使用列表而不是数组。问题是，我把它放在一个 numba njit 函数中，所以我需要精确一切。

目的是在双 for 循环中更改每个值。

Answer 1

numpy.zeros 的 shape 参数可以是任意长度的整数元组，因此您可以创建具有多个维度的 ndarray。

例如：

n = 5 # or any other number that you want
result = np.zeros((n,8,8), dtype=np.float32)

Create np.array filled with zero arrays