How to import a text file as a dictionary python

Question

I have a text file of the looks like this:

I'd like to make it a dictionary looking like this

graph = { "0" : ["1", "2", "3"],
      "1" : ["4", "0"],
      "2" : ["0", "1", "3"],
      "3" : ["0", "2", "4"],
      "4" : ["1", "3"]
    }

the file text list is a list of edges for a graph. I need to make a graph out of it without using any package. My final aim is to calculate the diameter and clustering coefficient. However, before starting I need to create the graph.

My attempt so far was:

d = {}
    with open("/Volumes/City_University/data_mining/Ecoli.txt") as f:
        for line in f:
           (key, val) = line.split()
           d[int(key)] = val
    for x in d:
    print (x)

Outcome:

Thanks

Justin O Barber · Accepted Answer · 2014-11-22 13:31:45Z

1

As one other possible option, you can also use defaultdict here:

from collections import defaultdict
d = defaultdict(list)
with open("/Volumes/City_University/data_mining/Ecoli.txt") as f:
    for line in f:
        key, val = line.split()
        d[key].append(val)
for k, v in d.items():
    print(k, v)

This saves you from having to check whether a key is already in d or not, and it also saves you a couple of lines.

answered Nov 22, 2014 at 13:31

Justin O Barber

11.6k2 gold badges43 silver badges45 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Arslan Over a year ago

this works better, I found also the 0 key in the result

Justin O Barber Over a year ago

Great. I'm glad you found it.

irrelephant · Accepted Answer · 2014-11-22 13:20:35Z

1

d = {}
with open("/Volumes/City_University/data_mining/Ecoli.txt") as f:
    for line in f:
       (key, val) = line.split()
       if key in d:
           d[key].append(val)
       else:
           d[key] = [val]
for x, v in d.items():
print x, v

Explanation:

Just make the values of d lists, and append to the lists.

answered Nov 22, 2014 at 13:20

irrelephant

4,1112 gold badges28 silver badges41 bronze badges

3 Comments

Arslan Over a year ago

awesome, thank you very much it seems it works. The only issue is that I can't find the key 0... How can I extract the print outcome to a txt file so that I can look better and see if the 0 key is there as well? Thanks again.

irrelephant Over a year ago

Please see this question.

Justin O Barber Over a year ago

@Andrea Sportelli You don't need to print to a file to check if a key is there. Simply add one more line to your code: print d.get('0').

Hackaholic · Accepted Answer · 2014-11-22 13:21:38Z

0

try this:

d = {}
with open("/Volumes/City_University/data_mining/Ecoli.txt") as f:
    for line in f:
       (key, val) = line.split()
       if key in d:
           d[key].append(val)
       else:
           d[key] = [val]
for x in d:
    print x,d[x]

if key is found in dictionary it will append the value else create a new pair

answered Nov 22, 2014 at 13:21

Hackaholic

19.8k6 gold badges59 silver badges77 bronze badges

Comments

K Adamu · Accepted Answer · 2014-11-22 13:50:01Z

0

import numpy as np
file_name='text_file.txt'
key_column=0

dat=np.genfromtxt(file_name,dtype=str)
d={i:[] for i in np.unique(dat[:,key_column])}

for row in dat:
    for key in d.keys():
        if row[key_column]==key :d[key].append(row[1])

print d

answered Nov 22, 2014 at 13:50

K Adamu

392 bronze badges

1 Comment

K Adamu Over a year ago

this might me good for when number of keys is too large, do two passes

Collectives™ on Stack Overflow

How to import a text file as a dictionary python

4 Answers 4

2 Comments

3 Comments

Comments

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

2 Comments

3 Comments

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related