2D NumPy数组+1D数组?

我猜你想把两个向量压缩成一个。如果是这样,举个例子 A 和 B 如下图所示

import numpy as np

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

你可以跑

list(map(list,zip(A,B)))

获得

[[array([1, 3, 6]), 1], [array([2, 2, 5]), 6]]