使用numpy组合两个n维数组

import numpy as np

# 创建两个n维数组
arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])

# 沿着行的方向连接两个数组
result = np.concatenate((arr1, arr2), axis=0)
print(result)
# 输出：
# [[1 2]
#  [3 4]
#  [5 6]
#  [7 8]]

# 沿着列的方向连接两个数组
result = np.concatenate((arr1, arr2), axis=1)
print(result)
# 输出：
# [[1 2 5 6]
#  [3 4 7 8]]

import numpy as np

# 创建两个n维数组
arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])

# 沿着新的轴堆叠两个数组
result = np.stack((arr1, arr2), axis=0)
print(result)
# 输出：
# [[[1 2]
#   [3 4]]
#
#  [[5 6]
#   [7 8]]]

# 沿着新的轴堆叠两个数组
result = np.stack((arr1, arr2), axis=1)
print(result)
# 输出：
# [[[1 2]
#   [5 6]]
#
#  [[3 4]
#   [7 8]]]