Вы можете инициализировать массив с помощью
rows = 2
cols = 5
a = np.zeros((rows, cols), float)
Это напечатает как:
array([[ 0., 0., 0., 0., 0.],
[ 0., 0., 0., 0., 0.]])
Затем вы можете получить доступ к вашим элементам следующим образом:
row_index = 1
col_index = 3
a[row_index, col_index] = 5
В результате получается:
array([[ 0., 0., 0., 0., 0.],
[ 0., 0., 0., 5., 0.]])
Как вы можете видеть в python, row_index исходит из 0 .. rows-1
.
Думаю, с этим знанием вы сможете реализовать свою формулу.
Дайте нам знать, как это происходит и где вам нужна дополнительная помощь.