两个列表

时间:2018-04-02 10:24:43

标签: python

我有以下文本文件:

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

但问题是,如果我转到下一行,我就无法分配其他元素。你知道如何解决这个问题吗?

"函数"只是我在别处定义的函数,它将字符串列表作为输入。

1 个答案:

答案 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