Pythondict
是一个非常有用的数据结构:
d = {'a': 1, 'b': 2}
d['a'] # get 1
有时您还想按值索引。
d[1] # get 'a'
实现这种数据结构的最有效方法是什么?有什么官方推荐的方法吗?
Pythondict
是一个非常有用的数据结构:
d = {'a': 1, 'b': 2}
d['a'] # get 1
有时您还想按值索引。
d[1] # get 'a'
实现这种数据结构的最有效方法是什么?有什么官方推荐的方法吗?
这是一个双向的类dict
,灵感来自从 Python 字典中的值中查找键,并修改为允许以下 2) 和 3)。
注意 :
bd.inverse
会自动更新bd
。bd.inverse[value]
始终是这样的列表。key
bd[key] == value
bidict
中的模块不同,这里我们可以有 2 个具有相同值的键,这非常重要。代码:
class bidict(dict):
def __init__(self, *args, **kwargs):
super(bidict, self).__init__(*args, **kwargs)
self.inverse = {}
for key, value in self.items():
self.inverse.setdefault(value,[]).append(key)
def __setitem__(self, key, value):
if key in self:
self.inverse[self[key]].remove(key)
super(bidict, self).__setitem__(key, value)
self.inverse.setdefault(value,[]).append(key)
def __delitem__(self, key):
self.inverse.setdefault(self[key],[]).remove(key)
if self[key] in self.inverse and not self.inverse[self[key]]:
del self.inverse[self[key]]
super(bidict, self).__delitem__(key)
使用示例:
bd = bidict({'a': 1, 'b': 2})
print(bd) # {'a': 1, 'b': 2}
print(bd.inverse) # {1: ['a'], 2: ['b']}
bd['c'] = 1 # Now two keys have the same value (= 1)
print(bd) # {'a': 1, 'c': 1, 'b': 2}
print(bd.inverse) # {1: ['a', 'c'], 2: ['b']}
del bd['c']
print(bd) # {'a': 1, 'b': 2}
print(bd.inverse) # {1: ['a'], 2: ['b']}
del bd['a']
print(bd) # {'b': 2}
print(bd.inverse) # {2: ['b']}
bd['b'] = 3
print(bd) # {'b': 3}
print(bd.inverse) # {2: [], 3: ['b']}
您可以通过以相反的顺序添加键值对来使用相同的字典本身。
d={'a':1,'b':2} revd=dict([reversed(i) for i in d.items()]) d.update(revd)
下面的代码片段实现了一个可逆(双射)映射:
class BijectionError(Exception):
"""Must set a unique value in a BijectiveMap."""
def __init__(self, value):
self.value = value
msg = 'The value "{}" is already in the mapping.'
super().__init__(msg.format(value))
class BijectiveMap(dict):
"""Invertible map."""
def __init__(self, inverse=None):
if inverse is None:
inverse = self.__class__(inverse=self)
self.inverse = inverse
def __setitem__(self, key, value):
if value in self.inverse:
raise BijectionError(value)
self.inverse._set_item(value, key)
self._set_item(key, value)
def __delitem__(self, key):
self.inverse._del_item(self[key])
self._del_item(key)
def _del_item(self, key):
super().__delitem__(key)
def _set_item(self, key, value):
super().__setitem__(key, value)
这种实现的优点是inverse
a 的属性BijectiveMap
又是 a BijectiveMap
。因此,您可以执行以下操作:
>>> foo = BijectiveMap()
>>> foo['steve'] = 42
>>> foo.inverse
{42: 'steve'}
>>> foo.inverse.inverse
{'steve': 42}
>>> foo.inverse.inverse is foo
True
不幸的是,评分最高的答案bidict
不起作用。
有三个选项:
子类 dict:您可以创建 的子类dict
,但要小心。您需要编写update
, pop
, initializer
,的自定义实现setdefault
。实现dict
不调用. __setitem__
这就是为什么评分最高的答案有问题。
从 UserDict 继承:这就像一个 dict,除了所有例程都被正确调用。它在一个名为data
. 您可以阅读Python 文档,或使用适用于 Python 3 的按方向列表的简单实现。很抱歉没有逐字包含它:我不确定它的版权。
从抽象基类继承:从collections.abc继承将帮助您获得新类的所有正确协议和实现。这对于双向字典来说是多余的,除非它也可以加密和缓存到数据库。
TL;DR -将此用于您的代码。阅读 Trey Hunner的文章了解详情。
像这样的东西,也许:
import itertools
class BidirDict(dict):
def __init__(self, iterable=(), **kwargs):
self.update(iterable, **kwargs)
def update(self, iterable=(), **kwargs):
if hasattr(iterable, 'iteritems'):
iterable = iterable.iteritems()
for (key, value) in itertools.chain(iterable, kwargs.iteritems()):
self[key] = value
def __setitem__(self, key, value):
if key in self:
del self[key]
if value in self:
del self[value]
dict.__setitem__(self, key, value)
dict.__setitem__(self, value, key)
def __delitem__(self, key):
value = self[key]
dict.__delitem__(self, key)
dict.__delitem__(self, value)
def __repr__(self):
return '%s(%s)' % (type(self).__name__, dict.__repr__(self))
如果多个键具有给定值,您必须决定要发生什么;给定对的双向性很容易被您插入的一些后来的对破坏。我实施了一种可能的选择。
例子 :
bd = BidirDict({'a': 'myvalue1', 'b': 'myvalue2', 'c': 'myvalue2'})
print bd['myvalue1'] # a
print bd['myvalue2'] # b
首先,您必须确保键值映射是一对一的,否则无法构建双向映射。
第二,数据集有多大?如果数据不多,就用2个单独的map,更新的时候同时更新。或者更好的是,使用像Bidict这样的现有解决方案,它只是 2 个 dicts 的包装,内置更新/删除。
但是如果数据集很大,并且不希望维护 2 个字典:
如果键和值都是数字,请考虑使用插值来近似映射的可能性。如果映射函数(及其反向函数)可以覆盖绝大多数键值对
,那么您只需要在映射中记录异常值即可。
如果大部分访问是单向的(键-> 值),那么增量构建反向映射以用时间换
空间是完全可以的。
代码:
d = {1: "one", 2: "two" }
reverse = {}
def get_key_by_value(v):
if v not in reverse:
for _k, _v in d.items():
if _v == v:
reverse[_v] = _k
break
return reverse[v]
更好的方法是将字典转换为元组列表,然后对特定的元组字段进行排序
def convert_to_list(dictionary):
list_of_tuples = []
for key, value in dictionary.items():
list_of_tuples.append((key, value))
return list_of_tuples
def sort_list(list_of_tuples, field):
return sorted(list_of_tuples, key=lambda x: x[field])
dictionary = {'a': 9, 'b': 2, 'c': 3, 'd': 4, 'e': 5}
list_of_tuples = convert_to_list(dictionary)
print(sort_list(list_of_tuples, 1))
输出
[('b', 2), ('c', 3), ('d', 4), ('e', 5), ('a', 9)]