pythonThe truth value of an array with more than one element is ambiguous. Use a.any() or a.all()の意味

Question

前提・実現したいこと

python3系を用いて関数を定義し、h(n)のnの値によってif文で分岐したいです

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

27行目のi==1がエラー出て今います。
エラーメッセージ
The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

該当のソースコード

import numpy as np
import math
FN=401
nmax=(FN-1)/2
f=1
wlt=f*SampleT*2*math.pi
wct=math.pi-wlt
lim=wct/math.pi
n=np.arange(1,201)


def h(i):
    if i == 1:
       return lim
    else:
       return 1/((i-1)*math.pi)*math.sin((i-1)*wct)


print(h(n))

試したこと

i=nにし、np.i()==0としたり、i is 1としました

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

python3.8

Comments

[meg_]

コードは「コードの挿入」で記入してください。（インデントがくずれています）

[meg_]

質問のコードに「n==1」が見当たりませんが。

Answer 1

例えば、関数 h に対して

[0,1,2]

のような配列データを引数として渡した場合

if i == 1:

行の i == 1 の部分の結果は

[False, True, False]

のようなBool配列データとなります。

この場合に if文として True として処理するのか False として処理するのかの判断がつかないので明確にしなさい

というのが今回のエラーの内容です。

ですので

# Bool配列に１つでも Trueがあったら処理
if (i == 1).any():

や

# Bool配列のすべてが Trueの場合のみ処理
if (i == 1).all():

のような処理に置き換えることで明確するようにしてください。

そうではなく、配列の各要素毎に個別に h 関数を実行したいのであれば、呼び出し側で np.vectorize() などを使って関数をベクトル化すると良いかもしれません。

np.vectorize(h)(i)

Comments

[kisuAE_punirin]

np.vectorize(h)(i)の方で実現したいことができました。
i==1について大きな勘違いしていました。とても分かりやすかったです