从具有np个概率数组的2D np数组中选择随机行的方法可以通过以下步骤实现:
import numpy as np
probabilities = np.array([[0.2, 0.3, 0.5], [0.4, 0.1, 0.5], [0.1, 0.9, 0.0]])
cumulative_probabilities = np.cumsum(probabilities, axis=1)
random_number = np.random.rand()
selected_row = np.argmax(cumulative_probabilities > random_number, axis=1)
最终,selected_row将包含选择的随机行的索引。
这种方法可以根据每行的概率数组进行随机选择,概率越高的行被选择的概率也越大。这在一些应用场景中非常有用,例如根据权重选择样本、进行随机采样等。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云