python - 根据关键词匹配从文本中获取数据

Question

我有一个包含以下内容的文本文件:

name1 -1 -1 -1 1 -1 1 
...
...
...

nameN -1 -1 1 1 -1 1 

此外，我有一个函数可以随机返回文件中的名称之一。有没有办法将文本文档转换为可管理的Python数据类型？基本上，如果函数返回 name1，我想获取与“name1”关联的整数向量。

Answer 1

在 Python 中这非常简单:

• 将数据分成几行。
• 对于每一行，拆分为单元格。
• 第一个单元格是您的字典键，值是其他单元格。
• 将它们转换为 int。

完成!

data = """name1 -1 -1 -1 1 -1 1 
nameN -1 -1 1 1 -1 1"""

my_dict = {}

for line in data.split("\n"):
    cells = line.split()
    my_dict[cells[0]] = [int(x) for x in cells[1:]]

print my_dict
# {'nameN': [-1, -1, 1, 1, -1, 1], 'name1': [-1, -1, -1, 1, -1, 1]}
print my_dict['name1']
# [-1, -1, -1, 1, -1, 1]

如果你想读取一个文件:

my_dict = {}

with open('filename.txt') as data:
    for line in data:
        cells = line.split()
        my_dict[cells[0]] = [int(x) for x in cells[1:]]

print my_dict
# {'nameN': [-1, -1, 1, 1, -1, 1], 'name1': [-1, -1, -1, 1, -1, 1]}
print my_dict['name1']
# [-1, -1, -1, 1, -1, 1]