リストの数値がある範囲にあるときのデータを抽出したい

Question

前提・実現したいこと

python (pygame) を使ってドライブゲームのようなものを作っています。
背景の道が一定速度で動いてくるので、道から外れないように車を左右に動かす、というゲーム（？）です。
道はpygame.draw.linesを用いて描出しています。
車はpygame.draw.circleを用いて描出しています。
車は左右（x軸方向）にのみ動き、上下（y軸方向）には動きません。
道が車と同じy座標（y=400）になったときの道の左側のlineのx座標をファイルに出力したいです。

発生している問題・エラーメッセージ

道の描出の際にpygame.draw.linesを用いて各座標をリストで示しているためか、while文の9行目（if ys1==400:）でエラーが出てしまいます。
エラーの直し方、もしくは修正案等を教えていただきたいです。
よろしくお願いいたします。

File "<stdin>", line 9, in <module>
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

該当のソースコード

import pygame
from pygame.locals import *
import sys
import time
import datetime as dt
import random
import numpy as np
import heapq

delay_inputs = []

now=dt.datetime.now()
date = now.strftime('%Y%m%d-%H%M%S')
'output_{}.txt'.format(date)

start=time.time()

BLACK = (0, 0, 0)
RED = (255, 0, 0)
WHITE = (255, 255, 255)

pygame.init() #初期化
screen = pygame.display.set_mode((640, 500)) #画面サイズ
myclock = pygame.time.Clock()
myclock.tick(30)
pygame.display.flip()

xs = [random.randint(0,590)] #Lineの各点のx座標
for i in range(14):
    xs.append(
        random.randint(max(0, xs[-1]-50), min(590, xs[-1]+50))
    )

x_cir = xs[0]+25 #circleの初期位置指定

xsplus=np.array(xs)+50 #xs+50の値を先に定義

ys1=[450] #Lineの各点のy座標(前半)
for i in range(14):
    ys1.append(ys1[-1]-50)

velocity_y=5 #Lineの動く速さ


while True:
    process_time=time.time()-start
    time.sleep(0.1)
    ys1=np.array(ys1) + velocity_y #Lineの各点のy座標＋速さ
    screen.fill(BLACK)
    for x, y in zip(xs, ys1): #左側のlineの座標
        points_l = list(zip(xs, ys1))
    pygame.draw.lines (screen, WHITE, False, points_l)
    if ys1==400:
        with open('y1output_{}.csv'.format(date), 'a') as f:
            print('{},{},{}'.format(x_cir,xs,process_time),file=f) #時間を入れた
    pygame.draw.circle(screen, RED, (x_cir, 400), 5)
    pygame.display.update()
    for event in pygame.event.get():
        if event.type==pygame.QUIT:
            pygame.quit()
            sys.exit()
    pressed_key=pygame.key.get_pressed()
    if (pressed_key[K_LEFT] and x_cir>0):
        x_cir -= 5
    if (pressed_key[K_RIGHT] and x_cir<640):
        x_cir += 5
    if pressed_key[K_ESCAPE]:
        pygame.quit()
        sys.exit()
    if time.time()-start>120:
        pygame.quit()
        sys.exit()

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

ys1に400を含むかどうかを調べたいという意味であれば、

if 400 in ys1:

とするのがいいでしょう。
他には以下のようなものがあるようです。
https://stackoverflow.com/questions/7088625/what-is-the-most-efficient-way-to-check-if-a-value-exists-in-a-numpy-array

Comments

[kappar]

ありがとうございます！
ちなみに、ys1が350から400のとき、というように指定範囲を広げる？場合はどのようにしたらよいでしょうか…？

[Penpen7]

np.anyがすぐ思いつきます。ys1に350以上, 400以下の要素が1つでもある場合、
if np.any((350 <= ys1) & (ys1 <=400)):
　なんらかの処理
でなんらかの処理を実行できます。

ちなみに、具体的にどの要素がヒットしたか調べるには
index = np.where((350 <= ys1) & (ys1 <=400))とすればインデックス(何番目か)のリストが帰ってきます。
具体的な値ではないことに注意してください。
具体的な値を取り出すには
ys1[index]とすればリスト形式で帰ってきます。

[kappar]

ありがとうございます！
すみません、重ねてもう1つだけ質問なのですが
if np.any(350 <= ys1 <=400):
をしたら
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()
というエラーが出てしまうのですが、これってなぜですか…？

[Penpen7]

すみません。多分条件が二つ合った時はよくなかったはずです...
少しお待ちください...

[Penpen7]

一つの条件にまとめるためにnp.absで絶対値を使ったほうが良さげですね。
np.any(np.abs(ys1-375)<=25)でどうでしょうか。
条件は350<=ys1<=400と同値です。

[kappar]

if np.any((350 <= ys1)&(ys1 <= 400)):
にしたらできました…。

[kappar]

教えていただいたものでもできました！ありがとうございます！

[Penpen7]

よかったです！
普段通りに条件を一気に書いてしまうと、ダメなんですよね
私もなんどもそこでつまづくので全然覚えられていないです笑
条件一個ずつにしてandとかorで結べばいけますので！
その旨、コメント修正しておきました。