Получение списка переменных из функции в python

Question

Я работаю над проектом городского паблика c space.

С помощью следующей функции я получил массив, связанный с растровой полосой городской среды.

#array generation function
def shade_array (index):
    altitude = df.altitude[index]
    azimut = df.azimut[index]
    shade = es.hillshade(elevation, azimuth = azimut, altitude = altitude)
    return shade

Теперь я хочу повторить функцию для разных дней в часе из кадра данных. Я запускаю следующее для -l oop:

for i in df.index:
    shade_array(i)

Дело в том, что мне нужно сохранить информацию в списке переменных, например:

array_list = [array_01, array_02,..., array_365]

в скрестить и сделать некоторые преобразования в полученных значениях. Как я могу добраться до него?

Answer 1

Создайте список , пока вы выполняете эту функцию:

array_list = [shade_array(i) for i in df.index]

Answer 2

Если вы хотите сохранить данные в виде списка переменных, просто используйте метод списка append(), который добавит переданный объект shade_array(i) в существующий список array_list:

array_list = []  
for i in df.index:
    array_list.append(shade_array(i))

или еще более кратким подходом может быть использование понимания списка , как в ответе usr2564301.