python - 使用子字符串打印出多个列表

Question

我有一个包含列表的字典，我想创建一个可以使用“substring”查找列表的函数。因此，如果列表的名称是“星期一要做的事情”，您只需键入“星期一”即可找到该列表。

这是我目前所拥有的。

字典。

list_dict = {}
list_dict["Things to do on Monday"] = ['clean', 'workout', 'buy food']
list_dict["Grocery list"] = ['oranges', 'apples', 'milk']
list_dict["Monday outfit"] = ['jeans', 't-shirt', 'shoes']

一个函数。

def substring_function(substring):
    for key in list_dict.keys():
        if substring in key:
            return list_dict[key]

然后，我举个例子:

print(substring_function("Monday"))

问题是，如您所见，我在两个列表中都有“星期一”一词，但我不知道如何让它返回两个列表。

我想我的 for 循环有问题。

谁能帮帮我？非常感谢!

Answer 1

您可以通过将 return 语句替换为 yield 语句来使函数成为生成器:

def substring_function(substring):
    for key in list_dict.keys():
        if substring in key:
            yield list_dict[key]

这样 list(substring_function("Monday")) 将返回匹配列表。