Odtnhj

So i have the below text as an input file:

A B 1
A C 2
B C 1
B D 3
B E 2
C D 1
C E 2
D E 4
D F 3
E F 3

and i want to store all these values in a nested dictionary in python:

 'A': 'B': 1, 'C': 2,
'B': 'C': 1, 'D': 3, 'E': 2,
'C': 'D': 1, 'E': 2,
'D': 'E': 4, 'F': 3,
'E': 'F': 3 

This is my code which is reading an input file and trying to convert it into a nested dictionary

inputfile = open("input.txt", "r")
lines = inputfile.readlines()
edges = 
for line in lines:
 edges.append(line.split())
d = 
nn=

for i in edges:
 nn.update(i[1]:i[2])
 d[i[0]] = nn
 index+=1

print(d)

This is my incorrect output

'A': 'B': '1', 'C': '1', 'D': '1', 'E': '4', 'F': '3', 'B': 'B': '1', 'C': '1', 'D': '1', 'E': '4', 'F': '3', 'C': 'B': '1', 'C': '1', 'D': '1', 'E': '4', 'F': '3', 'D': 'B': '1', 'C': '1', 'D': '1', 'E': '4', 'F': '3', 'E': 'B': '1', 'C': '1', 'D': '1', 'E': '4', 'F': '3'

I am looking for a correct if else condition which will update the dict items when the key is same and append if it is different. Thanks in advance.

You can do:

di=
with open(fn) as f_in:
 for line in f_in:
 li=line.split()
 di.setdefault(li[0],)[li[1]]=li[2]

>>> di
'A': 'B': '1', 'C': '2', 'B': 'C': '1', 'D': '3', 'E': '2', 'C': 'D': '1', 'E': '2', 'D': 'E': '4', 'F': '3', 'E': 'F': '3'

You can use recursion with collections.defaultdict:

import collections
def group(d):
 _d = collections.defaultdict(list)
 for a, *b in d:
 _d[a].append(b)
 return a:group(b) if len(b) > 1 else dict(b) if len(b[0]) > 1 else b[0][0] for a, b in _d.items()

content = [i.strip('n').split() for i in open('filename.txt')]
final_result = group(content)

Output:

'A': 'B': '1', 'C': '2', 'B': 'C': '1', 'D': '3', 'E': '2', 'C': 'D': '1', 'E': '2', 'D': 'E': '4', 'F': '3', 'E': 'F': '3'

This will also work on larger input samples:

s = """
 A B C D
 A C D E
 A H I F
 B D T G
 B F E H
 B U F A
 """
content = [i.split() for i in filter(None, s.split('n'))]
print(group(content))

Output:

'A': 'B': 'C': 'D', 'C': 'D': 'E', 'H': 'I': 'F', 'B': 'D': 'T': 'G', 'F': 'E': 'H', 'U': 'F': 'A'