如何从h5数据集中访问随机指数？

Question

我有一些h5数据，我想通过使用一些随机生成的指数进行采样。然而，如果指数的增长顺序不正确，那么努力就会失败。是否可以从h5数据集中选择随机生成的指数

以下是引用错误的MWE：

import h5py
import numpy as np
arr = np.random.random(50).reshape(10,5)
with h5py.File('example1.h5', 'w') as h5fw:
    h5fw.create_dataset('data', data=arr)

random_subset = h5py.File('example1.h5', 'r')['data'][[3, 1]]

# TypeError: Indexing elements must be in increasing order

我可以对指数进行排序，但这样我们就失去了随机性成分

Answer 1

一种可能的解决方案是对所需索引进行预排序，如下所示：

idx = np.sort([3,1])
random_subset = h5py.File('example1.h5', 'r')['data'][idx]