Numpy генерирует случайный массив на основе значений другого массива

Question

У меня есть 2 numpy массивов, a и b. Здесь:

a = np.random.randint(501, size=100)

Как можно случайным образом сгенерировать массив b размера 100 в векторизованном виде, так что:

1. b[i] > a[i] для всех значений i.
2. Каждое значение b лежит между [0, 501).

Answer 1

numpy поддерживает параметры массива. Вы можете использовать a в качестве нижней границы:

>>> rng = np.random.default_rng()
>>> 
>>> a = rng.integers(501,size=10)
>>> 
>>> a
array([ 82,  95, 463, 367, 257, 296, 449, 473, 202, 468])
>>> 
>>> b = rng.integers(a,501)
>>> 
>>> b
array([104, 153, 476, 376, 366, 391, 458, 474, 470, 499])