水平堆棧2d numpy陣列和3d numpy陣列

Question

我正在生成用於機器學習的特征數據集，我有一個二維的numpy數組X，其中X.shape =（n，d）-n個樣本，d個特征。

現在，我生成了一個帶有一鍵編碼的新功能-f，其中f.shape =（n，1，k）-n個樣本，k個標簽。

對我來說，將這個新特征添加到現有特征數據集中的最佳方式是什么？

Answer 1

一熱點向量的第二維是多余的，因此您可以將其刪除並將f用作形狀(n, k)的二維數組。
您將執行以下操作：

new_data = np.concatenate((X, f.squeeze()), axis=1)

squeeze()函數從您的數組中刪除所有1維（即f.squeeze().shape == (n, k) 。

干杯