python - 我如何获得给定两个单词的所有组合列表？

Question

我试图想出一种方法，以所有可能(但常见)的方式(如小写、大写和大写)获取诸如"is"之类的单词列表。然后我发现你不能在这个函数中放两个词(“sup”和“hello”)。有没有一种方法可以使用此功能将所有单词放在一个列表中，或者应该重新开始？

def case_insensetive(text) :
    insensetive_string = [text.lower(),text.upper(),text.capitalize()]
    print (insensetive_string)

try :
   case_insensetive("sup","hello")
except :
   raise Exception ("you screwed something")

Answer 1

这是 map() 的典型工作功能。

此外，在您的函数中使用 return 而不是 print。

def case_insensetive(text):
    insensetive_string = [text.lower(),text.upper(),text.capitalize()]
    return insensetive_string


words = ['yes', 'hello']

r = list(map(case_insensetive, words))

print(r)

输出:

[['yes', 'YES', 'Yes'], ['hello', 'HELLO', 'Hello']]

如果您想要单个列表，而不是嵌套列表:

flat_list = [item for sublist in r for item in sublist]
print(flat_list)

['yes', 'YES', 'Yes', 'hello', 'HELLO', 'Hello']