Python 中是否有一个函数可以从 n 维 numpy 数组中采样并返回每次绘制的索引。如果不是,人们将如何定义这样一个函数?
例如:
>>> probabilities = np.array([[.1, .2, .1], [.05, .5, .05]])
>>> print function(probabilities, draws = 10)
([1,1],[0,2],[1,1],[1,0],[0,1],[0,1],[1,1],[0,0],[1,1],[0,1])
Run Code Online (Sandbox Code Playgroud)
我知道这个问题可以用一维数组以多种方式解决。但是,我将处理大型 n 维数组,不能仅仅为了进行一次绘制就重新调整它们的形状。