Python - 来自txt的列表列表

时间:2018-02-28 00:01:42

标签: python list append

我是编程新手,我正在寻找一种方法来根据txt文件中每行的第一个数字对列表进行排序。在txt文件中,数字如下:

0 1

0 2

1 2

1 3

1 0

2 0

2 3

...

我想创建一个像这样的列表列表:[[1,2],[2,3,0],[0,3,] ...]我尝试追加和排序但我可以& #39;得到我想要的答案。 谢谢你的帮助!

1 个答案:

答案 0 :(得分:0)

您可以使用itertools.groupby

import itertools
from operator import itemgetter

with open("your/file/path.txt") as f:
    lines = sorted(f.readlines())

groups = [[int(ss[1:]) for ss in grp] for _, grp in itertools.groupby(lines, itemgetter(0))]