首页 > 解决方案 > 如何从以相同字母开头和结尾的单词创建单词部分词典

问题描述

我正在尝试编写一个脚本 - read_dict(dictionary) - 它将 .txt 文件作为参数,并给出文件中每个单词的单词部分的字典。这些部分是每个单词的第一个和最后一个字母,以及其余的字母。例如,如果文件如下:

===dictionary.txt===
quack  qk
quick qk
going gg
gathering gg
quirk qk
quicken qn

输出应该是:

{'qk':{'uac','uic'},'gg':{'oin','atherin'},'qn':{'uicke'}}

我写了这个:

def outside(word):
    a = word.strip()[0]
    b = word.strip()[-1]
    out_word = a + b
    return out_word


def inside(word):
    a = word.strip()[1:-1]
    return a


def read_dict(dictionary):
    a = {}
    with open(dictionary, 'r') as text:
        data = text.readlines()
        for i in data:
            a[outside(i)] = inside(i)
    return a

但我的输出是:

{'qk':'uac','gg':'oin','qn':'uicke'}

它只保存第一个单词。我也找不到一种方法来收集所有在集合中具有相同字母的内部(单词),然后使用适当的键(例如“qk”)将它们添加到字典中。

标签: pythonpython-3.xdictionary

解决方案


正如@Ch3steR 所说,这可以通过collections.defaultdict. 将您的代码修改为:

from collections import defaultdict

def read_dict(dictionary):
    a = defaultdict(set)
    with open(dictionary, 'r') as text:
        data = text.readlines()
        for i in data:
            a[outside(i)].add(inside(i))
    return a

如果您不想使用任何外部库,可以执行以下操作:

def read_dict(dictionary):
    a = {}
    with open(dictionary, 'r') as text:
        data = text.readlines()
        for i in data:
            key = outside(i)
            if key in a:
                a[key].add(inside(i))
            else:
                a[key] = {inside(i)}
    return a

通过比较这两个代码片段,您还可以了解它的collections.defaultdict作用以及它如何让您编写更少的代码。


推荐阅读