使用列表理解的函数只返回 1 个元素......？

如何解决使用列表理解的函数只返回 1 个元素......？

我正在开发一个函数，它以字典为键，以电子邮件域为键，以人员列表为值。

代码如下：

def email_list(domains):
    emails = []
    for users in domains.values():
      for user in users:
        emails.append(user)
        if user in domains["gmail.com"]:
            return [x + "@gmail.com" for x in emails]
        elif user in domains["yahoo.com"]:
            return [x + "@yahoo.com" for x in emails]
        elif user in domains["hotmail.com"]:
            return [x + "@hotmail.com" for x in emails]
        
    

print(email_list({"gmail.com": ["clark.kent","diana.prince","peter.parker"],"yahoo.com": ["barbara.gordon","jean.grey"],"hotmail.com": ["bruce.wayne"]}))

输出只返回：

['[e-mail]']

有人能解释一下发生了什么以及如何解决吗？

解决方法

我想这就是你想要的

def email_list(domains):
    emails = []
    for users in domains.values():
        for user in users:
            if user in domains["gmail.com"]:
                emails.append(user + "@gmail.com")
            elif user in domains["yahoo.com"]:
                emails.append(user + "@yahoo.com")
            elif user in domains["hotmail.com"]:
                emails.append(user + "@hotmail.com")
        
    return emails

print(email_list({"gmail.com": ["clark.kent","diana.prince","peter.parker"],"yahoo.com": ["barbara.gordon","jean.grey"],"hotmail.com": ["bruce.wayne"]}))

输出：

['[e-mail]','[e-mail]','[e-mail]','[e-mail]','[e-mail]','[e-mail]']

,

一个 return 关键字标志着函数的结束。在您的代码中，只要执行 return ... 语句，函数就会结束。所以，其余的代码没有用。

你可以做一个紧凑的解决方案。
您可以执行以下操作：

def email_list(domains):
    emails = [[user+"@"+server  for user in users ] for server,users in domains.items()]
    return emails

或者更紧凑

def email_list(domains):
    return [[user+"@"+server  for user in users ] for server,users in domains.items()]

如果您希望在一个列表中包含所有电子邮件地址，您可以：

return [user+"@"+server for server,users in domains.items() for user in users]

,

让我们来看看代码。当您第一次遇到 emails.append(user) 时，user 是“clark.kent”。接下来，你遇到了这个块：

        if user in domains["gmail.com"]:
            return [x + "@gmail.com" for x in emails]

此时，emails 是 ['clark.kent']，因此您的列表推导结果为 ['[e-mail]']，然后您将其返回。这里的问题是你一看到第一个用户就返回。

这是我的方法：

def email_list(domains):
    emails = [
        f"{user}@{domain}"
        for domain,users in domains.items()
        for user in users
    ]
    return emails

print(
    email_list(
        {
            "gmail.com": ["clark.kent","hotmail.com": ["bruce.wayne"],}
    )
)

请注意，我根本不需要测试域。相反，我只是使用域（密钥）来构建电子邮件地址。另请注意，我的列表推导式是双循环的，但它们与您的代码是相同的循环。