Python Set Literals

现有3种方式创建set()

>>> def f():
...     return set([1, 2, 3])
...
>>> def h():
...     return set((1, 2, 3))
...
>>> def g():  #set Literals
...     return {1, 2, 3}

>>> f()
set([1, 2, 3])
>>> h()
set([1, 2, 3])
>>> g()
set([1, 2, 3])

效率对比

>>> import timeit
>>> min(timeit.repeat(f))
0.675529956817627
>>> min(timeit.repeat(h))
0.5531120300292969
>>> min(timeit.repeat(g))
0.35515809059143066

对比结果

可见set Literals完胜

原理

为了了解其中的原因，分析下生成set()的原理

>>> dis.dis(f)
  2           0 LOAD_GLOBAL              0 (set)
              3 LOAD_CONST               1 (1)
              6 LOAD_CONST               2 (2)
              9 LOAD_CONST               3 (3)
             12 BUILD_LIST               3
             15 CALL_FUNCTION            1
             18 RETURN_VALUE
>>> dis.dis(h)
  2           0 LOAD_GLOBAL              0 (set)
              3 LOAD_CONST               4 ((1, 2, 3))
              6 CALL_FUNCTION            1
              9 RETURN_VALUE
>>> dis.dis(g)
  2           0 LOAD_CONST               1 (1)
              3 LOAD_CONST               2 (2)
              6 LOAD_CONST               3 (3)
              9 BUILD_SET                3
             12 RETURN_VALUE 

分析

f()需要载入全局函数set，把三个元素放入栈中，然后调用set()函数，就生成了set()

h()也需要载入全局函数set，不是把三个元素载入栈，而是把一个元组常量放入栈，然后调用set()函数，就生成了set()

g()是直接把三个元素放入栈，然后就生成了set()

总结

尽管这里生成set()的方式对性能的提升很小，set literals不用花费时间调用函数处理中间数据，并且这种写法是非常漂亮的，所以建议用set literals的方式