是否可以在 Python 中创建字典理解(用于键)?
如果没有列表推导,您可以使用如下内容:
l = []
for n in range(1, 11):
l.append(n)
我们可以将其缩短为列表推导:l = [n for n in range(1, 11)]
。
但是,假设我想将字典的键设置为相同的值。我可以:
d = {}
for n in range(1, 11):
d[n] = True # same value for each
我试过这个:
d = {}
d[i for i in range(1, 11)] = True
但是,我在 for
上得到了一个 SyntaxError
。
另外(我不需要这部分,只是想知道),您可以将字典的键设置为一堆不同的值,如下所示:
d = {}
for n in range(1, 11):
d[n] = n
这可以通过字典理解来实现吗?
d = {}
d[i for i in range(1, 11)] = [x for x in range(1, 11)]
这也会在 for
上引发 SyntaxError
。
有 dictionary comprehensions in Python 2.7+,但它们的工作方式与您尝试的方式不同。就像列表推导式一样,它们创建了一个 new 字典;您不能使用它们将键添加到现有字典中。此外,您必须指定键和值,当然如果您愿意,您可以指定一个虚拟值。
>>> d = {n: n**2 for n in range(5)}
>>> print d
{0: 0, 1: 1, 2: 4, 3: 9, 4: 16}
如果要将它们全部设置为 True:
>>> d = {n: True for n in range(5)}
>>> print d
{0: True, 1: True, 2: True, 3: True, 4: True}
您似乎要求的是一种在现有字典上一次设置多个键的方法。没有直接的捷径。您可以像已经展示的那样循环,也可以使用字典推导式来创建具有新值的新字典,然后执行 oldDict.update(newDict)
将新值合并到旧字典中。
您可以使用 dict.fromkeys
类方法...
>>> dict.fromkeys(range(5), True)
{0: True, 1: True, 2: True, 3: True, 4: True}
这是创建所有键映射到相同值的字典的最快方法。
但是do not use this with mutable objects:
d = dict.fromkeys(range(5), [])
# {0: [], 1: [], 2: [], 3: [], 4: []}
d[1].append(2)
# {0: [2], 1: [2], 2: [2], 3: [2], 4: [2]} !!!
如果您实际上需要初始化所有键,defaultdict
也可能有用:
from collections import defaultdict
d = defaultdict(True)
要回答第二部分,您需要的是 dict-comprehension:
{k: k for k in range(10)}
您可能不应该这样做,但您也可以创建一个 dict
的子类,如果您覆盖 __missing__
,它的工作方式有点像 defaultdict
:
>>> class KeyDict(dict):
... def __missing__(self, key):
... #self[key] = key # Maybe add this also?
... return key
...
>>> d = KeyDict()
>>> d[1]
1
>>> d[2]
2
>>> d[3]
3
>>> print(d)
{}
d = defaultdict(lambda: True)
的情况下,不需要 lambda,因为 True 是(或不应该)是可变的。
>>> {i:i for i in range(1, 11)}
{1: 1, 2: 2, 3: 3, 4: 4, 5: 5, 6: 6, 7: 7, 8: 8, 9: 9, 10: 10}
我真的很喜欢@mgilson 评论,因为如果您有两个可迭代对象,一个对应于键,另一个对应于值,您还可以执行以下操作。
keys = ['a', 'b', 'c']
values = [1, 2, 3]
d = dict(zip(keys, values))
给予
= {'a':1,'b':2,'c':3}
考虑这个使用字典理解计算列表中单词出现的例子
my_list = ['hello', 'hi', 'hello', 'today', 'morning', 'again', 'hello']
my_dict = {k:my_list.count(k) for k in my_list}
print(my_dict)
结果是
{'again': 1, 'hi': 1, 'hello': 3, 'today': 1, 'morning': 1}
在元组列表上使用 dict() ,此解决方案将允许您在每个列表中具有任意值,只要它们的长度相同
i_s = range(1, 11)
x_s = range(1, 11)
# x_s = range(11, 1, -1) # Also works
d = dict([(i_s[index], x_s[index], ) for index in range(len(i_s))])
d = dict(zip(i_s,x_s))
相同
列表推导的主要目的是在不更改或破坏原始列表的情况下基于另一个列表创建新列表。
而不是写
l = []
for n in range(1, 11):
l.append(n)
或者
l = [n for n in range(1, 11)]
你应该只写
l = range(1, 11)
在上面的两个代码块中,您正在创建一个新列表,遍历它并返回每个元素。这只是创建列表副本的一种昂贵方式。
要获取一个新字典,其中所有键都基于另一个字典设置为相同的值,请执行以下操作:
old_dict = {'a': 1, 'c': 3, 'b': 2}
new_dict = { key:'your value here' for key in old_dict.keys()}
您收到 SyntaxError 是因为当您编写
d = {}
d[i for i in range(1, 11)] = True
您基本上是在说:“将我的键 'i for i in range(1, 11)' 设置为 True”并且“i for i in range(1, 11)”不是有效键,它只是一个语法错误。如果 dicts 支持列表作为键,你会做类似的事情
d[[i for i in range(1, 11)]] = True
并不是
d[i for i in range(1, 11)] = True
但列表不可散列,因此您不能将它们用作 dict 键。
dictionary comprehension 与 list comprehension 非常相似,但我们在其末尾得到一个字典,因此我们需要分配键值对而不是仅分配值。
假设我们有一个用户列表,其中每个用户信息都存储在一个元组中。所以我们有一个包含四个用户元组的列表。在其中,他们有一个 ID、每个用户的唯一标识号、用户名和密码。
因此,我们要创建用户名到用户信息的映射。这是您经常会做的事情,尤其是当您在做诸如 Web 应用程序之类的事情时。
users = [
(0, "Bob", "password"),
(1, "code", "python"),
(2, "Stack", "overflow"),
(3, "username", "1234"),
]
username_mapping = {user[1]: user for user in users}
userid_mapping = {user[0]: user for user in users}
print(username_mapping)
"""
Why can this be helpful?
Well, imagine you know a user's username,and you want to get their information out.
You just access, let's say, "Bob," in your username_mapping, and you've got the information out.
"""
print(username_mapping["Bob"]) # (0, "Bob", "password")
# -- Can be useful to log in for example --
username_input = input("Enter your username: ")
password_input = input("Enter your password: ")
_, username, password = username_mapping[username_input]
if password_input == password:
print("Your details are correct!")
else:
print("Your details are incorrect.")
所以这是一个使用这个结构执行某种登录的例子,这个字典理解。
这真的很有帮助,因为它使您不必在此处执行另一个 for 循环,以确保您使用正确的用户名进行输入。
你不能像这样散列一个列表。试试这个,它使用元组
d[tuple([i for i in range(1,11)])] = True
dict.update
也可以像dict
构造函数一样接受可迭代的键值对dict.fromkeys()
。因此,要将所有值设置为True
,请使用dict.fromkeys(range(5), True)
。请注意,该值未复制,因此当您有一个可变值时,您可能希望避免这种情况;它将在所有密钥之间共享。{ n*2 : n for n in range(3) } => {0: 0, 2: 1, 4: 2}
。两者都可以在同一个表达式中完成:{ n*2 : n*3 for n in range(3) } => { 0: 0, 2: 3, 4: 6 }
。