создать массив переменного размера

Question

Я хотел бы создать простой массив, как показано ниже, за исключением того, что я хочу иметь возможность создавать массив переменной формы.Так что для приведенного ниже это будет n = 3.Есть ли отличный способ сделать это с NumPy или мне нужен цикл for.

выходные данные:

import numpy as np

np.array([1,0,0],[0,1,0],[0,0,1],[1,1,1],[0,0,0])

Answer 1

Скажем, вы хотите создать массив с именем d, имеющим row количество строк и col количество столбцов.Он также инициализирует все элементы массива с 0.

d = [[0 for x in range(col)] for y in range(row)]

Вы можете получить доступ к любому элементу I, j с помощью d [i] [j].