在python中为列表添加一个increaseKey

Question

假设一个包含字符串的列表 A 和一个偏移量 x。

A=["a","b","c"]
x 是整数
len(A) is int

我需要序列化并返回 json J 这样对于每个 n
J[x+n]=a[n]

我目前使用字典分配，但感觉它们更高效 而不是遍历整个列表。

答案需要是 O(1) 或矛盾。

谢谢。

dict={}

for i,x in enumerate(List):
    dict[i+offset]=x

Answer 1

如果您确实不想从列表中构造一个 dict，您可以使用依赖于 json 模块的实现细节的 hack：

class OffsetDict(dict):
    def __init__(self, lst, offset):
        self.lst = lst
        self.offset = offset
    def __nonzero__(self):
        return bool(self.lst)
    def iteritems(self):
        return enumerate(self.lst, self.offset)

A = ["a", "b", "c"]
d = OffsetDict(A, 5)
print json.dumps(d)

上面的代码打印

{"5": "a", "6": "b", "7": "c"}

因为构造OffsetDict 实例不会迭代列表，这部分的时间复杂度为 O(1)。当然，创建 JSON 输出不可避免地仍然是 O(n)。

该代码依赖于这样一个事实：json 将 dict 的每个子类视为 dict 并且仅调用 __nonzero__() 和 iteritems() 方法。不要指望该代码非常健壮。更强大的版本需要重新实现 dict 的所有方法。

在 Python 3.x 中，覆盖 items() 而不是 iteritems()。

Answer 2

给定以下变量（不要使用 dict/list 作为变量名！）：

offset = 5
A = ["a","b","c"]

您的示例代码可以编写为：

d = dict(zip(range(offset, offset+len(A)), A))

或使用 import itertools as it：

d = dict(it.izip(it.count(offset), A))

或（Python 2.6 及更高版本）：

d = dict(enumerate(A, offset))

在所有情况下d 现在是 {5: 'a', 6: 'b', 7: 'c'}。比较速度，第一个是最慢的，后两者没有显着差异。

Answer 3

dict 构造函数可以采用可迭代的键值对来构造字典。因此，您可以：

some_dictionary = dict((i + offset, x) for i, x in enumerate(some_list))

在较新的 Python（2.7 或 3.x）中，有用于 dict 理解的特殊语法：

some_dictionary = {i + offset: x for i, x in enumerate(some_list)}

Answer 4

为什么不在 A 前面添加一个长度为 offset 的列表？

result = [None] * offset + A