我試圖將以下兩個 np.arrays 組合成一個陣列：
prediction_score =
[0.99764085 0.26231623 0.07232302]
prediction_boxes =
[[282.25906 79.13187 420.98575 226.11221 ]
[109.91688 94.8121 333.07764 225.87985 ]
[340.3894 96.612015 601.4172 231.13196 ]]
組合陣列必須如下所示 [[i, pred, boxes],...]：
prediction_boxes =
[[1 0.99764085 282.25906 79.13187 420.98575 226.11221 ]
[1 0.26231623 109.91688 94.8121 333.07764 225.87985 ]
[1 0.07232302 340.3894 96.612015 601.4172 231.13196 ]]
我嘗試這樣做，但不幸的是它不起作用：
import numpy as np
i=1
for x in range(len(pred_scores)):
np.insert(pred_bboxes[x], 0, pred_scores[x])
np.insert(pred_bboxes[x], 0, i)
print(pred_bboxes)
有沒有辦法做到這一點？我嘗試了其他方法，但這些嘗試更糟。
uj5u.com熱心網友回復：
嘗試hstack：
np.hstack(([[1]]*len(pred_boxes), # classes
pred_scores[...,None], # scores
pred_boxes) # boxes
)
uj5u.com熱心網友回復：
Numpy 的 concatenate函式很好地處理了這個問題。嘗試類似：
output = np.concatanate((prediction_score.T, prediction_boxes), axis = 1)
uj5u.com熱心網友回復：
這應該有效： np.c_[prediction_score,prediction_boxes]
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/346654.html