Getting a list of variables from a function in python

Question

I'm working on a project related urban public space.

With the following function I got an array related to a raster band of an urban environment.

#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

Now I want to iterate the function for a different days an hour from a dataframe. I run the following for-loop:

for i in df.index:
    shade_array(i)

The point is that what I need is to store the info in a list of variables, like:

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

to cross and make some transformations in the values obtained. How can I get to it?

yacola · Accepted Answer · 2020-02-07 17:47:40Z

1

If you want to store data as a list of variables, just use list method append() that will append the passed object shade_array(i) into the existing list array_list:

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

or an even more concise approach may be to use list comprehension as in usr2564301's answer.

edited Feb 7, 2020 at 17:47

answered Feb 7, 2020 at 13:26

yacola

3,0332 gold badges13 silver badges30 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Jongware · Accepted Answer · 2020-02-07 14:19:05Z

1

Create the list while you are executing that function:

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

answered Feb 7, 2020 at 14:19

Jongware

22.6k8 gold badges56 silver badges104 bronze badges

Collectives™ on Stack Overflow

Getting a list of variables from a function in python

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related