Condition in list - Python

Question

I am working in python and I have this list

for i in range(0, len(state1)):
    observations = [{'state_transitions': [{'state': state1[i],
                                            'action': action[i], 
                                            'state_': state2[i]},],
                     'reward': 0},]

I would like to put the for clause inside the observations, something like this (but this is giving me an error):

observations = [
    for i in range(0, len(state1)):
        {'state_transitions': [{'state': state1[i],
                                'action': action[i],
                                'state_':state2[i]},],
        'reward': 0},]
print observations;

Can anyone help me with this?

I think you want list comprehension if I understand what you're saying. The first bit of code would overwrite observations with every iteration by the way. — Farhan.K
– Farhan.K, Commented Nov 25, 2016 at 13:40

Guillaume · Accepted Answer · 2016-11-25 13:41:34Z

3

What you are trying to accomplish (creating a list based on the results from a for loop) is called a list comprehension. Syntax is as follows: my_list = [do_something(item) for item in my_iterable].

Which gives:

observations = [
   { 
      'state_transitions': [
          { 'state': state1[i], 'action': action[i], 'state_':state2[i] },
       ],
        'reward': 0
    } for i in range(0, len(state1))
]
print(observations)

answered Nov 25, 2016 at 13:41

Guillaume

6,1293 gold badges28 silver badges48 bronze badges

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

Comments

jsbueno · Accepted Answer · 2016-11-25 13:46:42Z

1

Python does feature a for clause that can go inside list declarations - but it is placed after your expression - so this would work:

observations = [{ 
    'state_transitions': [{ 
         'state': state1[i], 'action': action[i], 'state_':state2[i]
          }],
    'reward': 0
   } for i in range(0, len(state1))
]

Besides that, Python's for is designed for higer elvel interations - if what interests you is each item in the sequence, and not the variable i itself, you can use the zip call to yield you one item of each sequence:

observations = [{
    'state_transitions': [{ 'state': st1, 'action': act, 'state_':st2}],
    'reward': 0
    } for st1, act, st2 in zip(state1, action, state2)
]

edited Nov 25, 2016 at 13:46

answered Nov 25, 2016 at 13:40

jsbueno

114k11 gold badges159 silver badges239 bronze badges

Collectives™ on Stack Overflow

Condition in list - 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