python边表转邻接表

import numpy as np


ft = open("figure.txt", "r")
b = {}

lines = ft.readlines()
for line in lines:
    a = list(line.rstrip().split("\t"))


    if a[0] in list(b.keys()):
        if a[1] not in b[a[0]]:
            b[a[0]].append(a[1])
    else:
        b[a[0]] = []
        b[a[0]].append(a[1])

    if a[1] in list(b.keys()):
        if a[0] not in b[a[1]]:
            b[a[1]].append(a[0])
    else:
        b[a[1]] = []
        b[a[1]].append(a[0])
ft.close()

# print(b)

f = open('tet.txt','w')
for bkey in list(b.keys()):
    f.write(bkey)
    for bkeyItem in b[bkey]:
        f.write("\t")
        f.write(bkeyItem)
    f.write("\n")
f.close()