垂直堆叠二维数组

A = [
    [1, 2],
    [3, 4]
]

B = [
    [5, 6],
    [7, 8]
]

C = [
    [1, 2],
    [3, 4],
    [5, 6],
    [7, 8]
]

import numpy as np

# 定义两个二维数组
A = np.array([[1, 2], [3, 4]])
B = np.array([[5, 6], [7, 8]])

# 垂直堆叠
C = np.vstack((A, B))

print(C)

[[1 2]
 [3 4]
 [5 6]
 [7 8]]

A = np.array([[1, 2], [3, 4]])
B = np.array([[5, 6]])

# 填充B使其与A行数相同
B_padded = np.pad(B, ((0, A.shape[0]-B.shape[0]), (0, 0)), 'constant')

C = np.vstack((A, B_padded))
print(C)

import h5py

# 假设我们有很多小数组需要堆叠
arrays = [np.random.rand(1000, 1000) for _ in range(10)]

# 使用HDF5存储中间结果
with h5py.File('stacked_arrays.hdf5', 'w') as f:
    stacked = f.create_dataset('stacked', (0, 1000), maxshape=(None, 1000), dtype='f')

    for arr in arrays:
        stacked.resize((stacked.shape[0] + arr.shape[0]), refcheck=False)
        stacked[-arr.shape[0]:] = arr