python - 如何将 vstack 代码转换为 for-loop (Python)？

Question

我目前有一个列表（list_arr），其中包含 8 个具有以下大小的 numpy 数组：

0. (6300, 6675, 3)
1. (5560, 6675, 3)
2. (5560, 6675, 3)
3. (5560, 6675, 3)
4. (6300, 6675, 3)
5. (5560, 6675, 3)
6. (5560, 6675, 3)
7. (5560, 6675, 3)

我想以 4 个为一组（例如，0-3 和 4-7）堆叠数组，以便输出数组的大小为（22980、6675、3）。这可以使用以下代码手动完成：

out1 = np.vstack((list_arr[0], list_arr[1], list_arr[2], list_arr[3]))

out2 = np.vstack((list_arr[4], list_arr[5], list_arr[6], list_arr[7]))

但是，我打算将这段代码升级为长度 = 116 的更长列表，而上述方法效率不高。有没有办法在 for 循环中做到这一点？

Answer 1

您可以对数组列表进行切片并将所需的切片数组传递给您的函数

检查以下示例代码：

arr_list = [1,2,3,4,5,6,7,8,9,0,12,33,45,66,77,88,23,21]
start = 0
for i in range(0,len(arr_list),4):
    if i == 0:
        continue
    print(arr_list[start:i])
    start = i
if start < len(arr_list):
    print(arr_list[start:])

输出是：

[1, 2, 3, 4]
[5, 6, 7, 8]
[9, 0, 12, 33]
[45, 66, 77, 88]
[23, 21]

如果您的数组的长度是 4 的精确倍数，那么您可以忽略最后一个 if 条件。

Answer 2

解决这个问题的方法不止一种。
让我向您展示一个 zip 示例，并让您探索更多选项。您也可以尝试使用 map、range、np.arange 和其他实用程序。

可以使用的for循环：

for i in zip(*[iter(list_arr)]*4):
    print(np.vstack(i))

另一种选择 - 带有 vsplit 的 for 循环：

for i in np.vsplit(list_arr, len(list_arr)//4):
    print(i)

与您一起试运行示例

list_arr = np.array([(6300, 6675, 3),
(5560, 6675, 3),
(5560, 6675, 3),
(5560, 6675, 3),
(6300, 6675, 3),
(5560, 6675, 3),
(5560, 6675, 3),
(5560, 6675, 3)])

for i in zip(*[iter(list_arr)]*4):
    print(np.vstack(i))

Output:
[[6300 6675    3]
 [5560 6675    3]
 [5560 6675    3]
 [5560 6675    3]]
[[6300 6675    3]
 [5560 6675    3]
 [5560 6675    3]
 [5560 6675    3]]

此代码适用于具有任意数量项目的数组，但输出的子列表将是 4 的倍数。