【问题标题】:Python: analyze a list comprehension with disPython:使用 dis 分析列表推导
【发布时间】:2018-03-31 10:32:57
【问题描述】:

最近,我在 SO 上有一个discussion(请参阅上下文)关于以下两段代码:

res = [d.get(next((k for k in d if k in s), None), s) for s in lst]

还有:

res = [next((v for k,v in d.items() if k in s), s) for s in lst]

两者都遍历列表lst 中的字符串s 并在字典d 中查找s。如果找到s,则返回相关值,否则返回s。我很确定第二段代码比第一段快,因为(对于每个s)在字典中没有查找,只是对(键,值)对的迭代。

问题是: 如何检查这是否真的发生在幕后?

我第一次尝试了dis 模块,但结果令人失望(python 3.6.3):

>>> dis.dis("[d.get(next((k for k in d if k in s), None), s) for s in lst]")
  1           0 LOAD_CONST               0 (<code object <listcomp> at 0x7f8e302039c0, file "<dis>", line 1>)
              2 LOAD_CONST               1 ('<listcomp>')
              4 MAKE_FUNCTION            0
              6 LOAD_NAME                0 (lst)
              8 GET_ITER
             10 CALL_FUNCTION            1
             12 RETURN_VALUE
>>> dis.dis("[next((v for k,v in d.items() if k in s), s) for s in lst]")
  1           0 LOAD_CONST               0 (<code object <listcomp> at 0x7f8e302038a0, file "<dis>", line 1>)
              2 LOAD_CONST               1 ('<listcomp>')
              4 MAKE_FUNCTION            0
              6 LOAD_NAME                0 (lst)
              8 GET_ITER
             10 CALL_FUNCTION            1
             12 RETURN_VALUE

如何获得更详细的信息?

编辑 正如@abarnert 在第一条评论中所建议的那样,我尝试了timeit 两种解决方案。我玩弄了以下代码:

from faker import Faker
from timeit import timeit

fake = Faker()

d = {fake.word():fake.word() for _ in range(50000)}
lst = fake.words(500000)

def f():return [d.get(next((k for k in d if k in s), None), s) for s in lst]
def g():return [next((v for k,v in d.items() if k in s), s) for s in lst]

print(timeit(f, number=1))
print(timeit(g, number=1))

assert f() == g()

也许我错过了什么,但令我惊讶的是,第一段代码 (f) 总是比第二段 (g) 快。因此,第二个问题:有人有解释吗?

编辑 2 以下是反汇编代码中最有趣的部分(稍加格式化以插入内部循环)。 对于f:

2           0 BUILD_LIST               0
          2 LOAD_FAST                0 (.0)
    >>    4 FOR_ITER                36 (to 42)
          6 STORE_DEREF              0 (s)
          8 LOAD_GLOBAL              0 (d)
         10 LOAD_ATTR                1 (get)
         12 LOAD_GLOBAL              2 (next)
         14 LOAD_CLOSURE             0 (s)
         16 BUILD_TUPLE              1
         18 LOAD_CONST               0 (<code object <genexpr> at 0x7ff191b1d8a0, file "test.py", line 2>)
         2           0 LOAD_FAST                0 (.0)
               >>    2 FOR_ITER                18 (to 22)
                     4 STORE_FAST               1 (k)
                     6 LOAD_FAST                1 (k)
                     8 LOAD_DEREF               0 (s)
                    10 COMPARE_OP               6 (in)
                    12 POP_JUMP_IF_FALSE        2
                    14 LOAD_FAST                1 (k)
                    16 YIELD_VALUE
                    18 POP_TOP
                    20 JUMP_ABSOLUTE            2
               >>   22 LOAD_CONST               0 (None)
                    24 RETURN_VALUE
         20 LOAD_CONST               1 ('f.<locals>.<listcomp>.<genexpr>')
         22 MAKE_FUNCTION            8
         24 LOAD_GLOBAL              0 (d)
         26 GET_ITER
         28 CALL_FUNCTION            1
         30 LOAD_CONST               2 (None)
         32 CALL_FUNCTION            2
         34 LOAD_DEREF               0 (s)
         36 CALL_FUNCTION            2
         38 LIST_APPEND              2
         40 JUMP_ABSOLUTE            4
    >>   42 RETURN_VALUE

对于g:

3           0 BUILD_LIST               0
          2 LOAD_FAST                0 (.0)
    >>    4 FOR_ITER                32 (to 38)
          6 STORE_DEREF              0 (s)
          8 LOAD_GLOBAL              0 (next)
         10 LOAD_CLOSURE             0 (s)
         12 BUILD_TUPLE              1
         14 LOAD_CONST               0 (<code object <genexpr> at 0x7ff1905171e0, file "test.py", line 3>)
         3           0 LOAD_FAST                0 (.0)
               >>    2 FOR_ITER                22 (to 26)
                     4 UNPACK_SEQUENCE          2
                     6 STORE_FAST               1 (k)
                     8 STORE_FAST               2 (v)
                    10 LOAD_FAST                1 (k)
                    12 LOAD_DEREF               0 (s)
                    14 COMPARE_OP               6 (in)
                    16 POP_JUMP_IF_FALSE        2
                    18 LOAD_FAST                2 (v)
                    20 YIELD_VALUE
                    22 POP_TOP
                    24 JUMP_ABSOLUTE            2
               >>   26 LOAD_CONST               0 (None)
                    28 RETURN_VALUE
         16 LOAD_CONST               1 ('g.<locals>.<listcomp>.<genexpr>')
         18 MAKE_FUNCTION            8
         20 LOAD_GLOBAL              1 (d)
         22 LOAD_ATTR                2 (items)
         24 CALL_FUNCTION            0
         26 GET_ITER
         28 CALL_FUNCTION            1
         30 LOAD_DEREF               0 (s)
         32 CALL_FUNCTION            2
         34 LIST_APPEND              2
         36 JUMP_ABSOLUTE            4
    >>   38 RETURN_VALUE

可以看到(同样由@abarnert 建议)g 的内部循环包含一些额外的成本:

  1. (隐藏)由d.items() 上的迭代器构建的 2-uples
  2. 一个 UNPACK_SEQUENCE 2 解包这些 2-uples,然后将 k 和 v 放入堆栈
  3. 两个STORE_FAST 从堆栈中弹出k 和v 以将它们存储在co_varnames 中。

在它最终加载k 之前将其与s 进行比较,如f。这个内部循环被迭代|lst|*|d| 并且似乎这些操作有所不同。

如果按照我的想法进行优化,d.items() 迭代器将首先将k 放入堆栈以测试k in s,然后,只有当k in s 为真时,才将v 放入YIELD_VALUE 的堆栈。

【问题讨论】:

  • 如果你想知道哪个更快,在你的头脑中处理它很好,深入研究编译的实现很好,但最重要的是拿出一些基准数据(这与你的问题)和timeit 不同的实现。如果你认为你可以证明一个更快,但事实证明它更慢,那么你的证明可能是错误的。 (或者你是对的,一个是对数线性的,另一个是线性的,但事实证明,对于最大数据来说,150 的常数因子比对数 N 更重要……)
  • @abarnert 我最感兴趣的是理论观点。正如您所指出的,timeit 的结果将取决于数据。但请参阅我对故事其余部分的编辑。
  • 您可能需要创建一个单独的问题来询问为什么 f 比 g 快(包括指向此的链接,以及您从 timeit 和阅读中发现的内容的摘要字节码)。这使问题集中在狭窄的范围内,这意味着理解算法问题但不知道字节码的人可以帮助回答它。此外,包含 Faker 的一些实现的 minimal reproducible example 意味着其他人可以在他们的系统上重复测试,逐步了解正在发生的事情等。
  • 我会做的。 Faker 是pypi.python.org/pypi/Faker
  • 我对性能差异的第一个猜测是 g 在不进行哈希表查找时所节省的成本被构建额外的 (k,v) 元组的成本所抵消。在非病态情况下,字典查找为 O(1),并且常量开销非常小(对于字符串键,除非使用已弃用的 C API 构建,散列被缓存在结构中)。

标签: python performance list-comprehension disassembly


【解决方案1】:

您已经获得了关于评估列表理解的代码的所有详细信息。

但是列表推导等同于创建然后调用一个函数。 (这就是它们拥有自己作用域的方式,因此它们不会,例如,将循环变量泄漏到外部作用域中。)因此,名为 &lt;listcomp&gt; 的自动生成函数是您真正想要查看的代码。

如果你想反汇编它——好吧,注意LOAD_CONST 0 说它正在加载一个&lt;code object &lt;listcomp&gt; at 0x7f8e302038a0?那就是你想要的。但是我们做不到,因为我们所做的只是为了反汇编而编译一个字符串,然后把结果扔掉,所以 listcomp 函数不再存在了。

但是用真实的代码很容易看出:

>>> def f():
...     return [next((v for k,v in d.items() if k in s), s) for s in lst]
>>> dis.dis(f)
  2           0 LOAD_CONST               1 (<code object <listcomp> at 0x11da9c660, file "<ipython-input-942-698335d58585>", line 2>)
              2 LOAD_CONST               2 ('f.<locals>.<listcomp>')
              4 MAKE_FUNCTION            0
              6 LOAD_GLOBAL              0 (lst)
              8 GET_ITER
             10 CALL_FUNCTION            1
             12 RETURN_VALUE

又是那个代码对象 const ——但现在它不仅仅是我们编译并立即丢弃的 const,它是我们可以访问的函数的一部分。

我们如何访问它?好吧,这在inspect 模块文档中有记录,这可能不是您第一次看到的地方。函数在其__code__ 成员中有一个代码对象,代码对象在其co_consts 成员中有一系列常量,我们正在寻找常量#1,所以:

>>> dis.dis(f.__code__.co_consts[1])
  2           0 BUILD_LIST               0
              2 LOAD_FAST                0 (.0)
        >>    4 FOR_ITER                32 (to 38)
              6 STORE_DEREF              0 (s)
              8 LOAD_GLOBAL              0 (next)
             10 LOAD_CLOSURE             0 (s)
             12 BUILD_TUPLE              1
             14 LOAD_CONST               0 (<code object <genexpr> at 0x11dd20030, file "<ipython-input-942-698335d58585>", line 2>)
             16 LOAD_CONST               1 ('f.<locals>.<listcomp>.<genexpr>')
             18 MAKE_FUNCTION            8
             20 LOAD_GLOBAL              1 (d)
             22 LOAD_ATTR                2 (items)
             24 CALL_FUNCTION            0
             26 GET_ITER
             28 CALL_FUNCTION            1
             30 LOAD_DEREF               0 (s)
             32 CALL_FUNCTION            2
             34 LIST_APPEND              2
             36 JUMP_ABSOLUTE            4
        >>   38 RETURN_VALUE

当然,您的列表推导式中嵌套了一个生成器表达式,并且您可能猜到,这也相当于创建然后调用一个生成器函数。但是该生成器函数的代码同样容易找到(如果输入起来更加繁琐):f.__code__.co_consts[1].co_consts[0]。

【讨论】:

  • 感谢您的回答和开明的 cmets。接受。
猜你喜欢
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 2016-02-13
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
相关资源
最近更新 更多