How can i convert this code to list comprehension in python
max_len = []
for word in fourier_dict:
word = fourier_dict[word]
for occur in word:
max_len.append(len(occur))
Im new to python and i have to convert this nested loop to a list comprehension which i cant figure out. A little help would be really appreciated.
max_len = [len(occur) for word in fourier_dict for occur in fourier_dict[word]]
Should work.
So first off, you don't actually use the key of the dict, so you can simplify to:
max_len = []
for word in fourier_dict.values():
for occur in word:
max_len.append(len(occur))
directly iterating the values (use .itervalues() if it's Py2 code to avoid a temporary list). From there, it's a simple transform to a listcomp, the value to "append" is on the far left, while the loops are laid out left to right, from outer-most to inner-most:
max_len = [len(occur) for word in fourier_dict.values() for occur in word]
You can try this:
s = [len(i) for a, b in fourier_dict.items() for i in b]
b before b exists. You also never use a, so you may as well just use .values() over .items().