python - 如何在 Python 中存储用于快速查找的一对一关系表？

Question

一个表的每两列之间是一对一的关系，比如说Ci <--> Cj .

如何存储这样的表以便快速查找？我将使用以下代码片段清楚地表达自己。

C1 = [1, 2, 3, 4]
C2 = ['a', 'b', 'c', 'd']
C3 = ['one', 'two', 'three', 'four']

# lookup, Ci --> Cj
idx = Ci.index(val)
corresponding_val = Cj[idx]

Dict 会是一个不错的选择。以只有两列的表为例，将表存储为dict，比如d[C1] = C2 .需要O(1)来自 C1至 C2 .但是来自C2至 C1 , 需要更长的时间。

Answer 1

如果您需要快速查找 C1、C2、C3 中的任何一个键，那么三个字典。每个的值是一个三元组。

all = zip( C1, C2, C3)
d1,d2,d3 = {},{},{}
for v in all:
   d1[ v[0]], d2[v[1]], d3[v[2]] = v,v,v

用法:

>>> d3['three']
(3, 'c', 'three')
>>> d1[1]
(1, 'a', 'one')
>>> d2['a']
(1, 'a', 'one')

这是仅访问一组元组数据的三个索引，因此它的效率与假设您每次快速查找需要一个哈希索引的效率差不多。

assert d1[1] is d2['a'] and d1[1] is d3['one']   

每一列只需要一个字典，因为访问的是整行。但是，假设任何列中都没有重复值。如果可能存在重复项，那么每个检索到的值都需要是一个行元组列表，而不仅仅是唯一的一个行元组。如果你需要这个，设置起来并不难:

C2=['odd','even','odd','even']
...
for v in all:
    d1.setdefault(v[0],[]).append(v)
    d2.setdefault(v[1],[]).append(v)
    d3.setdefault(v[2],[]).append(v)   

>>> d2
{'even': [(2, 'even', 'two'), (4, 'even', 'four')], 'odd': [(1, 'odd',  'one'), (3, 'odd', 'three')]}