二つのリストＸ，Ｙからそれぞれの要素x,yを含む関数を用いて特定の要素を抽出したい

Question

list_x = [0,1,..,99,100]
list_y = [0,1,..,99,100]
というリストがあったとします。それぞれのリストから任意の条件の要素番号を抜き出したい場合は

list_x = range(101)
list_y = range(101)

x_youso = [i for i, x in enumerate(list_x) if 52>=x>=12.5]

のように抜き出すことができますが、この二つのリストを一気に使い、例えば（以下のコードがエラーになることは承知ですが）

list_x = range(101)
list_y = range(101)

r_youso= [(i,j)for (i,j) , (x,y) in enumerate(list_x),(list_y) if 15**2>=(x**2+y**2)>=3**2]

のように抜き出すことはできないでしょうか？

Answer 1

zipしてenumerateすれば良いです。

>>> r_x = range(101)
>>> r_y = range(101)
>>>
>>> dst = [
...     i for i, (x, y) in enumerate(zip(r_x, r_y))
...     if (x + y) % 10 == 0
... ]
>>> print(dst)
[0, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75, 80, 85, 90, 95, 100]

Comments

[aine_]

ありがとうございます。