numpy - 在两个 numpy 数组之间随机选择值

Question

我有两个 numpy 数组:

left = np.array([2, 7])
right = np.array([4, 7])
right_p1 = right + 1

我想做的是

rand = np.zeros(left.shape[0])
for i in range(left.shape[0]):
  rand[i] = np.random.randint(left[i], right_p1[i])

有没有一种方法可以不使用 for 循环来做到这一点？

Answer 1

你可以试试:

   extremes = zip(left, right_p1)
   rand = map(lambda x: np.random.randint(x[0], x[1]), extremes)

这样您将得到一个 map 对象。如果你需要节省内存，你可以保持这种方式，否则你可以获得完整的 np.array 通过 list 转换，像这样:

rand = np.array(list(map(lambda x: np.random.randint(x[0], x[1]), extremes)))