Массив из функции в NumPy
Чтобы заполнить массив по индексам через функцию,
используют функцию fromfunction.
Она принимает
функцию от индексов и форму результата.
В функцию попадают не отдельные числа,
а массивы индексов по каждой оси.
Если для строки i и столбца
j нужна сумма индексов, каждая
ячейка получит i + j. Соберём
таблицу (2, 3) и выведем её:
import numpy as np
table = np.fromfunction(lambda i, j: i + j, (2, 3))
print(table) # выведет [[0. 1. 2.] [1. 2. 3.]]
Соберите таблицу формы (2, 2),
где в ячейке стоит сумма её индексов,
и выведите её.
Соберите одномерный массив длины
4, где значение равно индексу
i, и выведите его.