替代追加

Question

遵循python代码的好方法是什么：

          I = range(0,5000)     ## some list
          t = somevalue         ## for ex. 5    

          R = []     
          append1 = R.append     

          for i in I:      
              if i >= t:     
                 append1(1)
              else:
                 append1(0)

此代码工作正常，但我认为这不是一个好方法。附加方法也太慢了。

谢谢！

Answer 1

您可以使用列表理解：

R = [int(val>=t) for val in I]

Answer 2

列表理解似乎是你需要的：

>>> I = range(0,10)
>>> t = 3
>>> R = [1 if i>=t else 0 for i in I]
>>> R
[0, 0, 0, 1, 1, 1, 1, 1, 1, 1]

Answer 3

根据您的需要，这可能是另一种选择：

import numpy as np
I = np.arange(0,500)
t = 253
R = 1*(I >= t)

如果R需要是一个列表，您可以添加：R.tolist()