在循环中附加具有不相等行的numpy数组

Question

我想追加几个不同大小的数组。但是我不想将它们合并在一起，只是将它们存储在一个大型列表中。这是我的一个简化代码，它试图重现我的问题：

import numpy as np

total_wavel = 5

tot_values = []

for i in range(total_wavel):    
    
    size = int(np.random.uniform(low=2, high=7))
    values = np.array(np.random.uniform(low=1, high=6, size=(size,)))
    
    tot_values = np.append(tot_values,values)

示例输出：

array([4.88776545, 4.86006097, 1.80835575, 3.52393214, 2.88971373,
       1.62978552, 4.06880898, 4.10556672, 1.33428321, 3.81505999,
       3.95533471, 2.18424975, 5.15665168, 5.38251801, 1.7403673 ,
       4.90459377, 3.44198867, 5.03055533, 3.96271897, 1.93934124,
       5.60657218, 1.24646798, 3.14179412])

预期输出：

np.array([np.array([4.88776545, 4.86006097, 1.80835575, 3.52393214)], np.array([2.88971373,
       1.62978552, 4.06880898, 4.10556672]), np.array([1.33428321, 3.81505999,
       3.95533471, 2.18424975, 5.15665168, 5.38251801]), np.array([1.7403673 ,
       4.90459377, 3.44198867, 5.03055533], np.array([3.96271897, 1.93934124,
       5.60657218, 1.24646798, 3.14179412])])

或者

np.array([4.88776545, 4.86006097, 1.80835575, 3.52393214], [2.88971373,
       1.62978552, 4.06880898, 4.10556672],[1.33428321, 3.81505999,
       3.95533471, 2.18424975, 5.15665168, 5.38251801], [1.7403673 ,
       4.90459377, 3.44198867, 5.03055533], [3.96271897, 1.93934124,
       5.60657218, 1.24646798, 3.14179412])

提前致谢

Answer 1

在 for 循环中 tot_values.append(list(values))，在循环后 tot_np=np.array(tot_values)

在循环中附加具有不相等行的numpy数组

Append numpy array with inequal row in a loop

append

python-3.x

numpy-ndarray