Python：在一个文件行中检索两个单独的列表值

Question

我正在编写 python 中的代码，其中文件的一个特定行中有两个单独的值。我想将它们作为单独的部分检索到 matplotlib 的列表中。这是我到目前为止的代码：

with open('data.txt') as data_file:

    def process(line):
        line = line.rstrip(data_file)
        line = line.split('.')[1]
        line = line.split(',')
        return line


    x = list()
    y = list()

    counter = 0

    for line in data_file:
        if (counter == 3) or (counter == 4):
            result = process(line)
            x.append(int(result[0]))
            y.append(int(result[1]))
        counter += 1

print(x)
print(y)

错误是说：

line = line.rstrip(data_file)
TypeError: rstrip arg must be None or str

一个示例文件是：

hi
hi
67, 78
2345, 45677

谁能帮我解决这个错误，或者提供更好的方法来达到同样的结果。任何帮助表示赞赏！

非常感谢！

Answer 1

这是我能想到的：

import re

regex = r'[\d]{1,3}, [\d]{1,3}'
result = []
with open('sample.txt') as f:
    lines = f.readlines()
    for line in lines:
        match = re.findall(regex, line)
        if match != []:
            splitted = match[0].split(',')
            #the values are mapped to a list containing floating point numbers
            mapped = list(map(float, splitted))
            #and then are appended to a list that will contain all of
            #the lines that have the numbers on it
            result.append(mapped)

    print(result)
    #this is how you could access each line in result
    for list in result:
        print(list)

输出

[[67.0, 78.0], [25.0, 18.0]] #result is a list containing all lines that have the pattern <number>, <number>
[67.0, 78.0] #the first line that matches the pattern
[25.0, 18.0] #the second one

这使用正则表达式来查找最多 3 位数字（但您可以将其更改为您想要的任何数字），匹配模式 <number>, <number>

如果它与模式匹配，它会拆分, 处的两个数字，创建一个包含这两个值的列表并将它们附加到结果列表中

希望对你有帮助。

有任何问题欢迎提出。

编辑

我用这个作为一个示例文件来举例说明：

hi
hi
67, 78
hi again
25, 18