有没有一种方法可以使用numpy.random.randint初始化向量的一部分，然后初始化另一个向量？

Question

我需要初始化数组（索引）的前三个元素，其值介于1和120之间，接下来的3个元素的值介于1和140之间 这是我尝试过的

import numpy as np
indice=np.zeros((1,6))
indice[0:3]=np.random.randint(1,121,3)
indice[3:6]=np.random.randint(1,141,3)

，我得到这个错误： 回溯（最近通话最近）： 文件“”，第1行，位于 ValueError：无法将输入数组从形状（3）广播到形状（1,6）

Answer 1

您可以执行以下操作：

import numpy as np
indices1 = np.random.randint(1,121, (1,3))
indices2 = np.random.randing(1, 141, (1,3))
indices = np.concatenate((indices1, indices2), axis=1)