Есть ли метод для инициализации только части вектора, а затем другого, используя numpy .random.randint?

Question

Мне нужно инициализировать первые 3 элемента моего массива (индекс) со значениями от 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)

и Я получаю эту ошибку: Traceback (последний вызов был последним): файл "", строка 1, в ValueError: не удалось передать входной массив из shape (3) в shape (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)