Question

我需要阅读像这样的文本文件

bruce
chung
bruce@outlook.com
8893243
costarricense
divisa del informe colones
['nokia', 5277100.0, 'china']
['samsung', 10554200.0, 'turkey']
['apple', 52771000.0, 'argentina']

但是我需要只读取文件中的列表并将它们分配给一个新变量。

我用过这个

data =[ast.literal_eval(x) for x in open("test.txt")]
registros = data

在我添加以下用户的信息之前，它运作良好。

有没有办法只阅读文本中的列表？

Answer 1

您可以使用str.startswith()检查字符串是否以'['开头，并且只有在{true}时才使用ast.literal_eval()。示例 -

with open("test.txt") as f:
    registros = [ast.literal_eval(x) for x in f if x.startswith('[')]

演示 -

>>> s = """bruce
... chung
... bruce@outlook.com
... 8893243
... costarricense
... divisa del informe colones
... ['nokia', 5277100.0, 'china']
... ['samsung', 10554200.0, 'turkey']
... ['apple', 52771000.0, 'argentina']""".splitlines()
>>> import ast
>>> registros = [ast.literal_eval(x) for x in s if x.startswith('[')]
>>> registros
[['nokia', 5277100.0, 'china'], ['samsung', 10554200.0, 'turkey'], ['apple', 52771000.0, 'argentina']]

对于评论中的更新要求 -

我怎样才能只阅读用户信息并使用此方法将其分配给变量？

假设任何不是列表的行都是用户信息的一部分

你可以使用一个简单的for循环，如果该行不以'['开头，则继续附加到变量，如果是，则在其上使用ast.literal_eval()，并将其附加到{{ 1}}列表。

示例 -

registros

如果user_info总是只有6行（如评论中所示），你也可以使用 -

import ast
with open("test.txt") as f:
    user_info = ''
    registros = []
    for line in f:
        if line.startswith('['):
            registros.append(ast.literal_eval(line))
        else:
            user_info += line

Answer 2

除非封装列表，否则您不打算将括号用于任何目的，这应该可行。

import re
import ast

matches = []
data = open("test.txt", 'r').read()
for listString in re.findall(r"\[.*\]", data):
  matches.append(ast.literal_eval(listString))

如何在Python中的文本文件中只读取列表？

2 个答案: