max関数のプロット?

Question

問題

x >= 0ならx、それ以外なら0を出力する関数をプロットしたくてmax()関数を使いたいのですが、pyplotの引数の渡し方でエラーとなります。

ダメコード

import numpy as np
from matplotlib import pyplot as plt

x = np.linspace(-10,10,100)
plt.plot(x, max(0, x))

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

下記のようにplot(x, y)のyを関数ではなく配列として渡せばエラーなくできますが、それはなぜなんでしょう?
上のコードがダメな理由を教えて下さい!

OKコード

import numpy as np
from matplotlib import pyplot as plt

x = np.linspace(-10,10,100)
y = [max(0, i) for i in x]
plt.plot(x, y)

出力↓

Answer 1

max関数は引数の中から最大値を探しているので、
max(0,1)だと0と1の比較になりますが、max(0, np.linspace(-10,10,100))だと0とnp.linspace(-10,10,100)それ自体の比較になっているのではないでしょうか。
y = [max(0, i) for i in x]だと、要素一個一個と0の比較だから成立しています。

Comments

[Scrutiny0335]

ありがとうございます!スッキリしました!