水平堆栈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) 。

干杯