1

您好，我想使用 for 循环从现有列表中创建两个新列表。我知道通过对列表进行切片可以得到该结果，但是我有一个包含 1000 多个条目的大列表，因此通过计算索引进行切片并不是很可靠。我正在学习python，所以可能还有其他我不知道的方法，如果有人能解决我的问题，我将不胜感激。

所以我有两个清单说：

 a=np.linspace(1,10,30)
 b=np.linspcae(0,6,30)

 a
array([  1.        ,   1.31034483,   1.62068966,   1.93103448,
         2.24137931,   2.55172414,   2.86206897,   3.17241379,
         3.48275862,   3.79310345,   4.10344828,   4.4137931 ,
         4.72413793,   5.03448276,   5.34482759,   5.65517241,
         5.96551724,   6.27586207,   6.5862069 ,   6.89655172,
         7.20689655,   7.51724138,   7.82758621,   8.13793103,
         8.44827586,   8.75862069,   9.06896552,   9.37931034,
         9.68965517,  10.        ])


 b
array([ 0.        ,  0.20689655,  0.4137931 ,  0.62068966,  0.82758621,
        1.03448276,  1.24137931,  1.44827586,  1.65517241,  1.86206897,
        2.06896552,  2.27586207,  2.48275862,  2.68965517,  2.89655172,
        3.10344828,  3.31034483,  3.51724138,  3.72413793,  3.93103448,
        4.13793103,  4.34482759,  4.55172414,  4.75862069,  4.96551724,
        5.17241379,  5.37931034,  5.5862069 ,  5.79310345,  6.        ])

所以我的代码是：

for i in a:
   if i >2.27-1. and i<2.27+1:
      print i

这给了我想要的元素。

1.31034482759
1.62068965517
1.93103448276
2.24137931034
2.55172413793
2.86206896552
3.1724137931

但我也想要bie中的对应元素

0.20689655
0.4137931
0.62068966
0.82758621
1.03448276
1.24137931
1.44827586

我如何从这里得到想要的结果。

4

1 回答 1

4

使用 numpy 时尽量避免 for 循环，它的速度要快得多，而且很重要。

mask = (a>1.27) & (a<3.27)
a[mask]
b[mask]

(a>1.27) & (a<3.27)只返回一个布尔数组，其中值是True满足条件时和False其他地方的值。通过将aand分配b给该数组，您将获得布尔数组索引所在的值True。

编辑：

list(a[(a>1.27) & (a<3.27)])
list(b[(a>1.27) & (a<3.27)])

如果你真的想返回一个列表。如果你在这个 for 循环中做其他事情，那么你会想要坚持使用 for 循环。

于 2013-04-23T11:43:01.137 回答