python - 如何从给定条件开始获取列表中的元素

Question

我的代码是:

 b=[1215316235,1215316234,1210216096,1210716053]
 for i in range(398):
    a=b[i]
    if a[0:5] == '12153':
        print(a)

我收到错误:

TypeError                                 Traceback (most recent call last)
<ipython-input-113-d3b104ffc744> in <module>()
  1 for i in range(398):
  2     a=b[i]
 ----> 3     if a[0:5] == '12153':
  4         print(a)

  TypeError: 'int' object is not subscriptable

如何获得预期的输出:

1215316235
1215316234

Answer 1

为了使您的逻辑正常工作，您需要在切片之前转换为 str。此外，您应该迭代值而不是范围，例如通过对列表进行切片。

更有效的方法是使用列表理解。以下是一些示例:

b = [1215316235, 1215316234, 1210216096, 1210716053]

import math

b = b*10000

%timeit [x[:5] == '12153' for x in map(str, b)]                # 22.4 ms per loop
%timeit [str(x)[:5] == '12153' for x in b]                     # 28.1 ms per loop
%timeit [x // 10**(int(math.log10(x))-4) == 12153 for x in b]  # 48.6 ms per loop