首页 > 解决方案 > 如何将 txt 文件转换为数组的哈希图

问题描述

我正在尝试使用如下格式的文本文件:

4
0 2 3 4
1 1 3 4
2 1 2 4
3 1 2 3

并将其解析为如下内容:

G = {1:[2,3,4], 2:[1,3,4], 3:[1,2,4], 4:[1,2,3]}

其中第一行是行数,每个后续行的第一个数字是数组在哈希图中的顺序。

标签: python

解决方案


data = """4
0 2 3 4
1 1 3 4
2 1 2 4
3 1 2 3""".splitlines()[1:]

# or
# with open("file.txt") as fp:
#     data = fp.readlines()[1:]

G = {i + 1: list(map(int, line.split()[1:])) for i, line in enumerate(data)}
print(G)

{1: [2, 3, 4], 2: [1, 3, 4], 3: [1, 2, 4], 4: [1, 2, 3]}


推荐阅读