实现你想要的最简单的方法是“按字面意思写”:
for value in dictionary.values():
del value[2:]
可以读作:
对于字典中的每个值,删除第二个之后的所有元素。
注意:
- 此解决方案不会创建新字典,它会就地修改原始字典
- 此解决方案不会创建新列表。它就地修改当前列表。
在某些情况下,您可能想要创建一个新字典和/或避免改变字典的值,在这种情况下,您应该避免使用del
而是重新分配切片值:
dictionary[key] = value[:2]
它似乎也是最有效的:
In [1]: %%timeit
...: a = {'A': [1,2,3,4,5], 'B': [1,2]}
...: for value in a.values():
...: del value[2:]
...:
1000000 loops, best of 3: 663 ns per loop
In [2]: %%timeit
...: a = {'A': [1,2,3,4,5], 'B': [1,2]}
...: for key, value in a.items():
...: a[key] = value[:2]
...:
1000000 loops, best of 3: 920 ns per loop
In [3]: %%timeit
...: a = {'A': [1,2,3,4,5], 'B': [1,2]}
...: for value in a.values():
...: if len(value) > 2:
...: del value[2:]
...:
1000000 loops, best of 3: 796 ns per loop
In [4]: %%timeit
...: a = {'A': [1,2,3,4,5], 'B': [1,2]}
...: a = {k:v[:2] for k,v in a.items()}
...:
1000000 loops, best of 3: 1.15 us per loop
如果大多数值的长度为 2 或更短,则if len(value) > 2:
仅在必要时添加 to slice 可能会略微提高性能。如果您希望大多数值长于 2,那么它只会增加开销。
关于你的第一个问题:做不作为del a["A"][3:len(a[2])]
输出。你得到一个. 如果您获得了该输出,您已经做了其他事情。[3, 4, [1, 2, 3]]
KeyError: 2