如何將數組連接到 numpy 中 nd 數組的第一個 position？

Question

我需要在 numpy 的 nd 數組的第一個 position 中添加額外的元素。

這是代碼：

tmp_boxes3d = [None,6]

tmp_scores = [None]

現在我需要將 tmp_scores 元素添加到 tmp_boxes3d 的第一個 position 中。

為了更好地理解

boxes = np.array([[1,2,3,4],[5,6,7,8]])
scores = np.array([0,0])

boxes_scores = [[0,1,2,3,4],[0,5,6,7,8]]  # result

所以我需要為 [None,6] 形狀做。

誰能幫我這個？

Answer 1

嘗試np.concatenate

boxes = np.array([[1,2,3,4],[5,6,7,8]])
scores = np.array([0,0])

np.concatenate((scores[:, np.newaxis],  boxes), axis=1)

array([[0, 1, 2, 3, 4],
       [0, 5, 6, 7, 8]])

或者， np.hstack

np.hstack((scores[:, np.newaxis],  boxes))