Как я могу сделать словарь из TXT-файла в Python

Question

У меня есть этот текстовый файл

A:{A:0, B:6, C:4, D:3, E:0, F:0, G:0}

B:{A:6, B:0, C:2, D:0, E:4, F:0, G:0}

C:{A:4, B:2, C:0, D:2, E:0, F:8, G:0}

D:{A:3, B:0, C:2, D:0, E:3, F:0, G:0}

E:{A:0, B:4, C:0, D:3, E:0, F:7, G:6}

F:{A:0, B:0, C:8, D:0, E:7, F:0, G:6}

G:{A:0, B:0, C:0, D:0, E:6, F:6, G:0}

---

titles = []


with open("graph.txt", "r") as file:

    for line in file:
        column=line.split(":")
        title=column[0]
        titles.append(title)

Мне нужно сделать словари для каждого названия, которое я получил как A, B, C

Answer 1

Это может быть достигнуто с помощью следующего примера:

import re
titles = []


with open("graph.txt", "r") as file:

  for line in file:
    if ':' in line:
      title=re.match(r"^(.*?):", line).groups()[0]
      dict_str= re.match("^.*?\{(.*?)\}", line).groups()[0]
      dictionary = {key:value for (key,value) in (item.strip().split(':') for item in dict_str.split(','))}
      titles.append({title: dictionary})
  for item in titles:
    print(item)

Это будет производить как:

{'A': {'A': '0', 'C': '4', 'B': '6', 'E': '0', 'D': '3', 'G': '0', 'F': '0'}}
{'B': {'A': '6', 'C': '2', 'B': '0', 'E': '4', 'D': '0', 'G': '0', 'F': '0'}}
{'C': {'A': '4', 'C': '0', 'B': '2', 'E': '0', 'D': '2', 'G': '0', 'F': '8'}}
{'D': {'A': '3', 'C': '2', 'B': '0', 'E': '3', 'D': '0', 'G': '0', 'F': '0'}}
{'E': {'A': '0', 'C': '0', 'B': '4', 'E': '0', 'D': '3', 'G': '6', 'F': '7'}}
{'F': {'A': '0', 'C': '8', 'B': '0', 'E': '7', 'D': '0', 'G': '6', 'F': '0'}}
{'G': {'A': '0', 'C': '0', 'B': '0', 'E': '6', 'D': '0', 'G': '0', 'F': '6'}}

Answer 2

Правильно отформатируйте каждую строку, и вы можете использовать ast.literal_eval. Я использовал регулярное выражение, чтобы найти каждый ключ и заменить его тем же ключом, заключенным в кавычки.

import ast
import re
KEY_PATTERN = re.compile(r'(\w+?):')
dics = []
with open('graph.txt') as f:
    for line in f:
        line = line.strip()
        if line:
            dic_str = "{" + KEY_PATTERN.sub(r'"\g<1>":', line) + "}"
            dics.append(ast.literal_eval(dic_str))
print(dics)

Может быть короче (хотя труднее читать):

import ast
import re
KEY_PATTERN = re.compile(r'(\w+?):')

with open('graph.txt') as f:
    dics = [ast.literal_eval("{" + KEY_PATTERN.sub(r'"\g<1>":', line) + "}") for line in f if line.strip()]
print(dics)

Вывод:

[{'A': {'A': 0, 'B': 6, 'C': 4, 'D': 3, 'E': 0, 'F': 0, 'G': 0}}, {'B': {'A': 6, 'B': 0, 'C': 2, 'D': 0, 'E': 4, 'F': 0, 'G': 0}}, {'C': {'A': 4, 'B': 2, 'C': 0, 'D': 2, 'E': 0, 'F': 8, 'G': 0}}, {'D': {'A': 3, 'B': 0, 'C': 2, 'D': 0, 'E':3, 'F': 0, 'G': 0}}, {'E': {'A': 0, 'B': 4, 'C': 0, 'D': 3, 'E': 0, 'F': 7, 'G': 6}}, {'F': {'A': 0, 'B': 0, 'C': 8, 'D': 0, 'E': 7, 'F': 0, 'G': 6}}, {'G': {'A': 0, 'B': 0, 'C': 0, 'D': 0, 'E': 6, 'F': 6, 'G': 0}}]

Если вы хотите, чтобы результат был всего один dict, то измените:

dics = []
# and
dics.append(ast.literal_eval(dic_str))

на

dics = {}
# and
dics.update(ast.literal_eval(dic_str))