Convert value of dictionary as a list?

Question

Let's say I was given with this text age_gender.txt

Female:18,36,35,49,19
Male:23,22,26,26,26

Here's the code I have so far

file = open("age_gender.txt")
   contents = file.read().splitlines()
   new_dictionary = dict(item.split(":") for item in contents)

return new_dictionary

When I call the function readfile() this is the output I get, however the list of value is still in quotation marks. How do you convert that each value into a list?

{'Female': '18,36,35,49,19', 'Male': '23,22,26,26,26'}

The output I want to achieve is something like this

{'Female': [18,36,35,49,19], 'Male': [23,22,26,26,26]}

Tony Babarino · Accepted Answer · 2016-03-17T00:08:03.180

>>> a
'Female:18,36,35,49,19,19,40,23,22,22,23,18,36,35,49,19,19,18,36,18,36,35,12,19,19,18,23,22,22,23'
>>> a.split(':')
['Female', '18,36,35,49,19,19,40,23,22,22,23,18,36,35,49,19,19,18,36,18,36,35,12,19,19,18,23,22,22,23']
>>> a.split(':')[1].split(',')
['18', '36', '35', '49', '19', '19', '40', '23', '22', '22', '23', '18', '36', '35', '49', '19', '19', '18', '36', '18', '36', '35', '12', '19', '19', '18', '23', '22', '22', '23']
>>> new_dictionary = dict({a.split(':')[0]:map(int,a.split(':')[1].split(','))}) 
>>> new_dictionary
{'Female': [18, 36, 35, 49, 19, 19, 40, 23, 22, 22, 23, 18, 36, 35, 49, 19, 19, 18, 36, 18, 36, 35, 12, 19, 19, 18, 23, 22, 22, 23]}

Applying that to Your code:

file = open("age_gender.txt")
   contents = file.read().splitlines()
   new_dictionary = dict()
   for item in contents:
       tmp = item.split(':')
       new_dictionary[tmp[0]] = list(map(int, tmp[1].split(',')))

return new_dictionary

Be aware that since this is Python3, `map` returns an iterator, not a list. If you want lists, you may want to wrap the `map` inside a `list` statement, i.e. `list(map(int, ..etc.))` — Steinar Lima, Mar 16 '16 at 23:14
Good point, I've checked the code only in python2. Thanks for that, fixed the answer.. — Tony Babarino, Mar 17 '16 at 00:09

Thomas Zumsteg · Answer 2 · 2016-03-16T22:59:16.150

2

You've go the basics down, the remaining steps are:

splitting the value on commas split(',')
Converting the string into an integer int(i)

Wrap these steps in a for loop and do it for every key/value pair the dictionary.

for key, value in new_dictionary.items():
    new_dictionary[key] = [int(i) for i in value.split(',')]

edited Mar 16 '16 at 22:59

answered Mar 16 '16 at 22:55

Thomas Zumsteg

118
6

Yes I do, @SvenMarnach thanks for the note. – Thomas Zumsteg Mar 16 '16 at 22:59

score 2 · Answer 3 · answered Mar 16 '16 at 23:10

Here is another way to do it using ast.literal_eval to convert the ages into a Python list. It has the advantage of supporting all basic data types, e.g. float, without explicit conversion:

from ast import literal_eval

with open('age_gender.txt') as f:
    d = {gender: literal_eval(ages) for gender, ages in (line.split(':') for line in f)}

This will produce a dictionary with tuples as values:

{'Male': (23, 22, 26, 26, 26), 'Female': (18, 36, 35, 49, 19)}

If you really, really need lists, you can convert the tuples:

with open('age_gender.txt') as f:
    d = {gender: list(literal_eval(ages)) for gender, ages in (line.split(':') for line in f)}

{'Male': [23, 22, 26, 26, 26], 'Female': [18, 36, 35, 49, 19]}

Very good. i like this approach. – Sandeep Mar 16 '16 at 23:28 — Sandeep, Mar 16 '16 at 23:28

score 1 · Answer 4 · answered Mar 16 '16 at 22:48

1

You need to split this dictionary value by "," and then map it to int:

s['Female'] = map(int, s['Female'].split(','))

answered Mar 16 '16 at 22:48

ig-melnyk

2,679
2
20
33

Convert value of dictionary as a list?

4 Answers4