python - 按出现顺序获取 numpy 数组索引

Question

我正在使用train_test_split。我的训练集 X[] 是一个文件路径数组。然后我有另一个数组 y[]，它由一个热编码标签组成。它们通过数组行索引相关。因此，如果我通过 X，它看起来像这样:

索引路径

4、数据\djip2\DJIP2.5844MHz.10MSPS.fc32.2016-07-01_000000000001.npy

20、数据\taigentank\USRP-2_420GHz-1MSps-1MHztaigentank1_000000000000.npy

2、数据\866_300_1\USRP-866_300MHz-1MSps-1MHz_lte_1_000000000002.npy

y[] 看起来像这样:

索引标签

4, 00000001

20, 00000010

2, 01000000

这些数组在随机化后被传递到批处理生成器。在批处理生成器中，我需要确保每个 X 数组值都可以映射回 y[] 数组标签。

所以，我希望能够获取 X 数组索引，这些索引现在按随机顺序排列，例如:

2、路径

4、路径

20，路径

并按此顺序将它们传递给另一个函数。我需要索引，因为我需要传递路径及其关联的标签。有没有一种简单的方法可以用 numpy 来做到这一点？

Answer 1

一种解决方案可能是:

n = range(numberOfInstances)

它创建一个整数列表，例如[0,1,2,3,4...numberOfInstances-1]。然后随机排列列表

random.shuffle(n)

将此列表保存为 numpy 数组

n_np = np.array(n)

最后相应地重新排序你的数据和地面事实

y = y[n_np]
x = x[n_np]

应该以相同的方式重新排序两个数组。

我希望这有帮助:)!我有点困惑为什么你有一个无序的 ID 序列，但又以某种方式再次打乱......