1

こんにちは、for ループを使用して既存のリストから 2 つの新しいリストを作成したいと考えています。リストをスライスすることで結果が得られることはわかっていますが、1000 を超えるエントリを含む大きなリストがあるため、インデックスをカウントしてスライスすることはあまり信頼できません。私はpythonを学んでいるので、私が気付いていない他の方法があるかもしれませんので、誰かが私の問題を解決できれば幸いです.

だから私は言う2つのリストを持っています:

 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

これにより、 a から必要な要素が得られます。

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です。aこの配列にとを代入することにより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 に答える