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的方式