如何在包含相同首字母python的列表中查找单词

Question

编写一个函数common_start(word_list) ，它接受一个单词列表作为参数。 此函数必须返回一个新列表，其中包含以与列表中至少一个其他单词相同的字母开头的所有单词。

输入：['文件'，'编辑'，'查看'，'插入'，'格式']
输出：['文件'，'格式']

Answer 1

一个单行解决方案，如果你想

def common_start(input_list):
    return [word for word in input_list if sum([w.startswith(word[0]) for w in input_list]) > 1]

Answer 2

这应该对你有用。 在这种情况下，我忽略了重复的单词。

def common_start(word_list):
    output = set()

    for word in word_list:
        for first_letter in word_list:
            if word == first_letter:
                next
            elif word[0] == first_letter[0]:
                output.add(word)
    return list(output)

输入：['tara', 'file', 'edit', 'view', 'insert', 'format','test']

输出：['tara', 'file', 'format', 'test']