I am trying to convert the below to a dict comprehension
my_dict = {'a': None, 'b': None, 'c': ['1', '2', '3']}
new_dict = {}
for k, v in my_dict.items():
if not v:
new_dict[k] = None
else:
for item in v:
new_dict[f'{k}{item}'] = None
I'm trying to translate my dict to
new_dict = {'a': None, 'b': None, 'c1': None, 'c2': None, 'c3': None}
I'm fairly comfortable with basic list and dict comprehensions but struggling with this one, currently looking something like this but clearly I am a bit off on the syntax:
{k: None if not v else f'{k}{item}': None for item in v for k, v in my_dict.items()}
This abomination will do:
{fk: None
for k, v in my_dict.items()
for fk in ([k] if v is None else (k + fv for fv in v))}
If the value is None
, you just want the key.
If the value is not None
, you want a list of each value concatenated with the key.
Homogenise that to always returning a list, either of one key or multiple:
[k] if v is None else [k + fv for fv in v]
Then you're looking at a "simple" nested comprehension:
{k: None for k in [['a'], ['b'], ['c1', 'c2', 'c3']] for fk in k}
I was proud of my answer:
new_dict = dict(
sum(
[
[(k, None)] if not value else [(f"{k}{v}", None) for v in value]
for k, value in my_dict.items()
],
[],
)
)
... until I saw deceze's answer, which I still need to digest.
You can do this with the help of itertools.chain.from_iterable
:
>>> {
y: None
for y in itertools.chain.from_iterable(
k if not v else (f"{k}{x}" for x in v) for k, v in my_dict.items()
)
}
{'a': None, 'b': None, 'c1': None, 'c2': None, 'c3': None}
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.