0

这是针对机器学习问题(当然是在 Python 中)。

我有一个二维数组,行是点集,列是这些点的另一个一维值数组的索引。

data = [[1,3,2], [3,3,1], [5,1,2]]
# yes there are duplicates in the labels
labels = [2,8,9,8,8,9]

我需要创建一个二维数组,它是原始数据数组,但其中的值现在是索引表示的标签中的值。

new_data = [[8,8,9], [8,8,8], [9,8,9]]

显然,我可以用 for 循环来做到这一点。我在这里问,以防 numpy 或其他东西有这样做的电话。

4

1 回答 1

1

使用索引作为索引

np.array(labels)[np.array(data)]

高级整数)索引的输出是索引数组(data)的形状。

于 2021-09-09T16:43:15.373 回答