如何从列表中获取某些给定条件的元素

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

Answer 2

您可以先将b（整数列表）转换为字符串列表：

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

Answer 3

一旦你输入int到str，你可以查看指数。

b = [1215316235, 1215316234, 1210216096, 1210716053]
for i in map(str, b):
    if i[:5] == '12153':
        # this is a str; if you want an int change for spec above
        print(i) 

我个人会将它添加到一个函数中并删除那里的硬编码;虽然这完全取决于您以及您需要如何使用它。