Python3.x txt 文件与不同的分隔符到整数列表

发布于11月13日

我有一个包含整数的txt文件，其中一些用空格分隔，另一些用逗号分隔.

它看起来是这样的:

7,4,9,5,11,17,23,2,0,14,21,24,10,16,13,6,15,25,12,22,18,20,8,19,3,26,1

22 13 17 11  0

8  2 23  4 24

21  9 14 16  7

6 10  3 18  5

我想创建2个列表，1个列表与第一个CSV的列表和1个子列表与其他行的整数如下:

list_1 = [7,4,9,5,11,17,23,2,0,14,21,24,10,16,13,6,15,25,12,22,18,20,8,19,3,26,1]
list_2 = [[22, 13, 17, 11,  0], [8,  2, 23,  4, 24], [21,  9, 14, 16,  7], [6, 10,  3, 18,  5]]

我找不到我想要的结果，有谁能帮我吗？

我try 了不同的加载方法，但由于分隔符不同，总是出现错误:

 b=[[int(x) for x in i.split()] for i in open("file", "r").readlines()]

例如，无法读取逗号分隔值

推荐答案

try :

list_1 = []
list_2 = []

with open("your_file.txt", "r") as f_in:
    for line in map(str.strip, f_in):
        if line == "":
            continue
        if "," in line:
            list_1.extend(map(int, line.split(",")))
        else:
            list_2.append(list(map(int, line.split())))

print(f"{list_1=}")
print(f"{list_2=}")

打印:

list_1=[7, 4, 9, 5, 11, 17, 23, 2, 0, 14, 21, 24, 10, 16, 13, 6, 15, 25, 12, 22, 18, 20, 8, 19, 3, 26, 1]
list_2=[[22, 13, 17, 11, 0], [8, 2, 23, 4, 24], [21, 9, 14, 16, 7], [6, 10, 3, 18, 5]]