エラー:The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()について

Question

ステップ関数を、plt.plot（）の引数に入れ、グラフを描写してみたのですが、以下のようなエラーコードが出てしまいます。その原因と対処方法をご教示ください。よろしく、お願いいたします。

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

import numpy as np
import matplotlib.pyplot as plt

x=np.arange(-10,11)

def step_function(a):
    if a >0:
        return 1
    
    else:
        return 0

plt.plot(x,step_function(x))
plt.show()

# ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

Answer 1

原因は、a>0という配列にはTrueもあればFalseもあることです。

対処方法は以下の関数に変えることです。

def step_function(a):
    return (x>0).astype(np.int8)

結果は以下です。

Comments

[okahijiki]

ご回答、ありがとうございます。できましたー。
(x>0).astype(np.int8)をググりながら、自分なりに補助線を引いて、理解してみました。

０より大きいかどうか、True/Falseになっているのを、astype（)メソッドにてint型に変換した、といった感じでしょうか。