[英]How to get a dictionary key by value from a nested dictionary?
我有字典
game_objects = {
('wall', 0): {'position': (0, 0), 'passable': False, 'interactable': False, 'char': '#'},
('wall', 1): {'position': (0, 1), 'passable': False, 'interactable': False, 'char': '#'},
('player',): {'position': (1, 1), 'passable': True, 'interactable': True, 'char': '@', 'coins': 0},
('soft_wall', 11): {'position': (1, 4), 'passable': False, 'interactable': True, 'char': '%'}
}
我需要创建一个函数,该函数将从嵌套字典中按值获取字典键。 例如:
get_objects_by_coords((0, 1)) == [('wall', 1)]
get_objects_by_coords((1, 1)) == [('player',)]
get_objects_by_coords((2, 1)) == []
这是我所做的:
def get_objects_by_coords(position):
for position in game_objects.values():
if position in game_objects.values():
return game_objects.keys()
print(get_objects_by_coords((0, 0)))
但答案不正确
dict_keys([('wall', 0), ('wall', 1), ('player',), ('soft_wall', 11)])
我只需要这部分
[('wall', 1)]
那么,我该如何改进我的代码? 我知道这段代码很糟糕,但我只是在学习
您正在返回每个dict
键,而不是您找到的键。 我建议您查看dict.items
方法,该方法返回一对键值。 因此,您的函数可能如下所示:
def get_objects_by_coords(game_objects, position):
for key, value in game_objects.items():
if position == value.get('position', (None,)):
return key
get_objects_by_coords(game_objects, (0, 1)) # == ('wall', 1)
此外,如果您要在程序中多次迭代字典,我不会像您那样使用数据结构,因为尽管查看dict
键 i O(1),但对其进行迭代以获得一个值是 O(n)。 此外,我不会使用全局变量(您的函数使用game_objects
变量,即使它没有将其作为参数接收),因为这不是一个好习惯。
希望这有帮助! 祝你好运,编码愉快!
你是这个意思吗?
def get_objects_by_coords(position):
return [key for key, val in game_objects.items()
for k, v in val.items() if k == 'position' and v == position]
assert get_objects_by_coords((0, 1)) == [('wall', 1)]
assert get_objects_by_coords((1, 1)) == [('player',)]
assert get_objects_by_coords((2, 1)) == []
也许你的整个结构是错误的。 那这个呢?
game_objects = [
[
{
'type':'wall',
'item_no':0,
'passsable':False,
'interacable':False,
'char':'#'
},
{
'type':'wall',
'item_no':1,
'passsable':False,
'interacable':False,
'char':'#'
}
],
[
{
'type':'player',
'item_no':None,
'passable':True,
'interactable': True,
'char': '@',
'coins': 0
},
{
'type':None
},
{
'type':None
},
{
'type':None
},
{
'type':'soft_wall',
'item_no': 11,
'passable': False,
'interactable': True,
'char': '%'}
],
]
输出
game_objects[1][4]
{'type': 'soft_wall',
'item_no': 11,
'passable': False,
'interactable': True,
'char': '%'}
game_objects[0][1]
{'type': 'wall',
'item_no': 1,
'passsable': False,
'interacable': False,
'char': '#'}
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.