matplotlib绘制分段函数

可以使用`numpy`和`matplotlib`库来绘制分段函数，具体步骤如下：

1. 导入所需库

import numpy as np
import matplotlib.pyplot as plt

2. 定义分段函数

假设我们要绘制的分段函数为：

$$
f(x) = \begin{cases}
x^2, & x \leq 0 \\
2x, & 0 < x \leq 2 \\
4-x, & x > 2 \\
\end{cases}
$$

可以使用`numpy`库中的`where`函数来实现：

def piecewise(x):
    return np.where(x <= 0, x**2, np.where(x <= 2, 2*x, 4-x))

3. 绘制函数图像

使用`matplotlib`库中的`plot`函数可以绘制函数图像，具体步骤如下：

# 定义x轴的取值范围
x = np.linspace(-5, 5, 1000)

# 绘制函数图像
plt.plot(x, piecewise(x))

# 添加标题和坐标轴标签
plt.title("Piecewise Function")
plt.xlabel("x")
plt.ylabel("f(x)")

# 显示图像
plt.show()

完整代码如下：

import numpy as np
import matplotlib.pyplot as plt

def piecewise(x):
    return np.where(x <= 0, x**2, np.where(x <= 2, 2*x, 4-x))

x = np.linspace(-5, 5, 1000)
plt.plot(x, piecewise(x))
plt.title("Piecewise Function")
plt.xlabel("x")
plt.ylabel("f(x)")
plt.show()