从 Python 中的文本文件中读取多行变量

Question

我正在将文本文件读入python。 文本文件包含（大量）变量，变量名称在左侧以字符串形式给出，在右侧以等号 (=) 分隔值。 例如

Proc_Method = 2
Obs_Method = 4

只要变量的值在一行中给出，我就可以正确读出变量的值：

    namevalue = data.split('=') 
    name = namevalue[0].strip()
    value = namevalue[1].strip()

但是，如果变量分布在多行（即数组）上。 此代码仅将数组的第一行分配给变量，然后再转到下一个变量。 因此，如果我有以下形式的变量：

Corr_Mat = 10 0 0 
           20 10 0
           0 20 10  

上面的代码将声明该值等于 10 0 0，然后转到下一个变量。 有没有一种方法可以定义值，以便它采用以等号开头的所有行，并在文本文件中下一个相等之前的行结束？

Answer 1

使用这样的文件：

Proc_Method = 2
Obs_Method = 4
Corr_Mat = 10 0 0 
           20 10 0
           0 20 10
Proc_Method = 2

选项1

您可以遵循类似堆栈的方法，将新变量作为键放入字典variables中，并将所有以下行作为值附加到键中。

variables = {}

with open('file.txt') as f:
    for line in f:
        if '=' in line:
            # add a new key-value pair to the dictionary,
            # remember the current variable name
            current_variable, value = line.split('=')
            variables[current_variable] = value
        else:
            # it's a continued line -> add to the remembered key
            variables[current_variable] += line

结果：

{'Proc_Method ': ' 2\n',
 'Obs_Method ': ' 4\n',
 'Corr_Mat ': ' 10 0 0 \n           20 10 0\n           0 20 10\n'}

选项 2

或者，您可以读取整个文件并使用正则表达式来提取变量。

该模式搜索行的开头（前导^与re.MULTILINE组合），后跟任意数量的符号（ .+ ）后跟=后跟任意数量的不等号符号（ [^=]+ ) 后跟换行符。

import re 

with open('file.txt') as f:
    file_content = f.read()

chunks = re.findall(r'^.+=[^=]+\n', file_content, flags=re.MULTILINE)

结果：

['Proc_Method = 2\n',
 'Obs_Method = 4\n',
 'Corr_Mat = 10 0 0 \n           20 10 0\n           0 20 10\n',
 'Proc_Method = 2\n']

当然，这需要一些清理，并且只有在变量不包含任何可能无法保证的= s 时才有效。

Answer 2

试试这个代码。

params = {}

last_param = ''

for line in data.splitlines():
    
    line = line.strip()
    
    if '=' in line:
        sp = line.split('=')
        last_param = sp[0]
        params[last_param] = sp[1]
    else:
        params[last_param] += f'\n{line}'

print(params)

结果：

{'Proc_Method ': ' 2', 'Obs_Method ': ' 4', 'Corr_Mat ': ' 10 0 0\n20 10 0\n0 20 10'}

Answer 3

我得到了一个更清晰的结果，但可以说是更混乱的代码。 但我想有很多简单的方法可以让这段代码更干净......

文件输入：

Proc_Method = 2
Obs_Method = 4
Corr_Mat = 10 0 0 
           20 10 0
           0 20 10
test = 4
test2 = 4
woof = a b c 
           s sdfasd sdas 
           sda as a 

代码：

output = {}
with open ("E:\Coding\stackoverflow\input.txt", "r") as file:
    lines = file.readlines()
    previous_full_line = 0
    for line_number, line in enumerate(lines):
        line_content = line.strip()
        if "=" in line_content:
            namevalue = line_content.split('=') 
            name = namevalue[0].strip()
            values = namevalue[1].strip()
            values = values.split(" ")
            line_list = []
            for value in values:
                line_list.append(value)
            output[name] = line_list
            previous_full_line = line_number
        else:
            values = line_content.split(" ")
            new_list = []
            for value in values:
                new_list.append(value)
            output[name].extend(new_list)

print(output)

结果

{
   "Proc_Method":[
      "2"
   ],
   "Obs_Method":[
      "4"
   ],
   "Corr_Mat":[
      "10",
      "0",
      "0",
      "20",
      "10",
      "0",
      "0",
      "20",
      "10"
   ],
   "test":[
      "4"
   ],
   "test2":[
      "4"
   ],
   "woof":[
      "a",
      "b",
      "c",
      "s",
      "sdfasd",
      "sdas",
      "sda",
      "as",
      "a"
   ]
}

从这个输出结果我猜你可以随意使用数据。 知道字典“输出”中的所有属性和“键”都在一个数组中。