【发布时间】:2021-07-31 14:58:53
【问题描述】:
我尝试在 for 循环中将 numpy 数组(numpy_array 的倍数)添加到一个列表中,其中每个 numpy 数组可能具有不同的形状(但假设每个数组都有 (10, 64))。所以每个数组有 64 列,并且可能有尽可能多的行。我所做的如下:
l_arr = numpy_array[indices, :]
list_arr.append(l_arr.tolist())
那么,
numpy_from_list= np.array(list_arr)
但这似乎增加了额外的维度。例如,如果我将 3 个大小为 (10, 64) 的 numpy 数组添加到列表中,然后将列表转换为 numpy.我会得到(3, 10, 64),但我想要(30, 64)。
for i in range(3):
l_arr = numpy_array[indices, :]
list_arr.append(l_arr.tolist())
numpy_from_list= np.array(list_arr)
你能帮帮我吗?
【问题讨论】:
-
尝试使用numpy的np.stack()函数
-
@sehan2。谢谢,但我得到了相同的结果。
标签: numpy