我有以下文本文件:
text1 text2
# text2 text3 text4
# text5 text4 text6
text 3 text 4
# ....
...
我想有一个如下所示的数组列表,其中fu
dict[function([text1, text2])] = [[text2, text3, text4], [text5, text4, text6]].
想法是打开文件,逐行阅读
dict = {}
inputfile = open("text.txt","r")
for line in inputfile:
l=line.split()
if not line.startswith("#"):
#create a new key
else:
dict[key] = l
但问题是,如果我转到下一行,我就无法分配其他元素。你知道如何解决这个问题吗?
"函数"只是我在别处定义的函数,它将字符串列表作为输入。
答案 0 :(得分:0)
您可以使用collections.defaultdict
创建一个字典,其list
s作为您可以追加项目的值。
from collections import defaultdict
my_dict = defaultdict(list)
with open('text.txt') as f:
key = ''
for line in f:
if not line.startswith('#'):
key = line
# key = function(line.split())
continue
my_dict[key].append(line.strip('#').split())
print(my_dict)
输出:
defaultdict(<class 'list'>,
{'text1 text2\n': [['text2', 'text3', 'text4'],
['text5', 'text4', 'text6']],
'text3 text4\n': [['text21', 'text31', 'text41'],
['text51', 'text41', 'text61']]})
只需将key = line
行更改为您将密钥传递给的任何功能。
我的text.txt
文件包含:
text1 text2
# text2 text3 text4
# text5 text4 text6
text3 text4
# text21 text31 text41
# text51 text41 text61