Convert for loop to lambda/list comprehension in Python

Question

I have a list, loc_combinations, with a length of 91806 of unique ID pairs structured as so:

[(1,2), (1,3), 1,4)...(452, 454)]

I am trying to apply the same function distance_calculator to each pair in the list which returns a single value, the distance. I was able to get my answer using a for loop but was hoping someone could show me how to do it using Lambda and list comprehension.

Here is the for loop:

distance_list = []
for i in range(len(loc_combinations)): 
    distance_list.append(distance_calculator(id1 = loc_combinations[i][0], id2 = loc_combinations[i][1]))

SimonR · Accepted Answer · 2020-05-23 20:29:18Z

4

No need for a lambda function if you have a working function defined. Your code as a list comprehension looks like this:

distance list = [distance_calculator(x[0], x[1]) for x in loc_combinations]

answered May 23, 2020 at 20:29

SimonR

1,8241 gold badge7 silver badges12 bronze badges

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

1 Comment

Samwise Over a year ago

or distance_calculator(*x) if you're nasty

Jan Joswig · Accepted Answer · 2020-05-23 20:36:29Z

0

distance_list = [
    distance_calculator(id1=x, id2=y)
    for x, y in loc_combinations
]

answered May 23, 2020 at 20:36

Jan Joswig

7335 silver badges22 bronze badges

Collectives™ on Stack Overflow

Convert for loop to lambda/list comprehension in Python

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related