read text file lines and create a dictionary using python

Question

I have a following text file format

Id,    person, age,  city
ef12,  james,  23,   berlin  
yt34,  mary,   45,   pisa  
rt23,  john,   56,   barcelona

I want to generate a dictionary of below kind. Please help me out .

{ef12: {person:'james', age:'23',city:'berlin'},   
yt34: {person:'mary', age:'45',city:'pisa'},    
rt23: {person:'john', age:'23',city:'barcelona'},  

}

I am stuck at below

`import time
import sys

def getData():
    file = open('traffic.txt', 'r')
    data = file.readlines()
    myDic = {}
    #for line in data.split('\n'):
    for line in data:
        tmp = line.strip().split()
        #myDic[tmp[0]]= list(tmp[1])
        #print(tmp[2])
        myDic[tmp[0]] = {tmp[1],tmp[2],tmp[3],tmp[4],tmp[5]}
    file.close()
    return myDic
theNewDictionary = getData()
print(theNewDictionary)
`

Tell us what happens when you try to run this. I see that you try to access tmp[4] and tmp[5] but you only expect tmp to have 4 elements. — Paul Cornelius, Oct 28 '17 at 02:12
sorry, you are right about it ! i didn't put all text values, my bad — codegeeka, Oct 28 '17 at 04:05

Simon Mengong · Answer 1 · 2017-10-28T02:29:53.157

2

You have just to add the keys

def getData():
    file = open('traffic.txt', 'r')
    data = file.readlines()
    myDic = {}
    for line in data:
        tmp = [s.replace(' ', '') for s in line.strip().split(',')]
        myDic[tmp[0]] = {'person': tmp[1], 'age': tmp[2], 'city': tmp[3]}
    file.close()
    return myDic

edited Oct 28 '17 at 02:29

answered Oct 28 '17 at 02:18

Simon Mengong

2,625
10
22

Also the proper `split(",")` for each line. Usually `file.read().splitlines()` is more handy to create a list. – m3nda Oct 28 '17 at 02:21
@scharette the ouput is `{'ef12': {'person': 'james', 'age': '23', 'city': 'berlin'}, 'yt34': {'person': 'mary', 'age': '45', 'city': 'pisa'}, 'rt23': {'person': 'john', 'age': '56', 'city': 'barcelona'}}` I think it is the expected – Simon Mengong Oct 28 '17 at 02:31

score 1 · Answer 2 · answered Oct 28 '17 at 02:21

split by comma: split(',')
strip after split to remove spaces: [word.strip() for word in line.split(',')]
You only have 4 columns, so don't call tmp[4] and tmp[5] — it is IndexError if you do.
Name your keys in the dictionary: {'person': tmp[1], 'age': tmp[2], 'city': tmp[3]}

This means:

def getData():
    file = open('traffic.txt', 'r')
    data = file.readlines()
    myDic = {}
    for line in data:
        tmp = [word.strip() for word in line.split(',')]
        myDic[tmp[0]] = {'person': tmp[1], 'age': tmp[2], 'city': tmp[3]}
    file.close()
    return myDic

score 1 · Accepted Answer · edited Oct 30 '17 at 17:30

Another way is to read the rows from the csv and update the dictionary with each row:

dicty = {}

for row in csv.DictReader(open('a.csv')):
    dicty.update({
        row['Id']: {
            'person': row['person'],
            'age'   : row['age'],
            'city'  : row['city']
        }
    })

print(dicty)
# {'ef12': {'person': 'james', 'age': '23', 'city': 'berlin'},
#  'yt34': {'person': 'mary',  'age': '45', 'city': 'pisa'},
#  'rt23': {'person': 'john',  'age': '56', 'city': 'barcelona'}}

dicty.get('ef12')
# {'age': '23', 'city': 'berlin', 'person': 'james'}

read text file lines and create a dictionary using python

3 Answers3

Linked