how to do dict comprehension for multiple for and if loops in python?

user_15213571 0
Viewed 57

i have a code like below:

out={}
for each_val in values["data"]['values']:
    if each_val['value'].strip() != '':
       if 'frequency' in each_val:
           out[each_val['value']] = each_val['frequency']
       else:
           out[each_val['value']] = None

i am trying to make it one line with dict comprehension but not able to do it with 2 if loops inside for loop. Below is the code I tried but getting warnings and errors

out = {each_val['value']: each_val['frequency'] if 'frequency' in each_val and if each_val['value'].strip() != '' else None for each_val in values["data"]['values']}
dictionary-comprehension python-3.x dictionary
2 Answers

Remove if after and

out = {each_val['value']: each_val['frequency'] if 'frequency' in each_val and each_val['value'].strip() != '' else None for each_val in values["data"]['values']}
user_109686210

You should put the filter as an if clause in the comprehension. Also, use the dict.get method to default the value to None when a key is not found in the dict:

out = {
    each_val['value']: each_val.get('frequency')
    for each_val in values["data"]['values']
    if each_val['value'].strip() != ''
}
user_68909120
Related