Создание матрицы от 0 до nxn и преобразование ее в пары

Question

Может быть, я ошибаюсь, и я не знаю, как это искать.

Мне нужно получить список пар

(0,0) (0,1) (0,2) ... (0,999) ... (999,999)

Чтобы поместить эти значения в нейронную сеть и получить результат - просто вектор длины (999 * 999)

Затем я могу изменить его на (999,999) и получить карту выходных данных.

Таким образом, наивный подход заключается в использовании

for i in range(0, 999):
    for j in range(0, 999):
        list.add(predict(i,j))

Но невозможно использовать пакеты

Другой наивный подход -

numbers=[]
for i in range(0, 999):
    for j in range(0, 999):
        numbers.append([i,j])

, и это работает впартии.

Есть ли более элегантное решение?

Answer 1

Попробуйте:

target_list = np.array([(i,j) for i in range(1000) for j in range(1000)])

Этот print(target_list) выдаст желаемый список в формате массива:

[[  0   0]
 [  0   1]
 [  0   2]
 ...
 [999 997]
 [999 998]
 [999 999]]