用python绘制分段函数的代码

在Python中，我们可以使用matplotlib库来绘制分段函数。分段函数是由几个独立的部分组成，每个部分都有自己的数学表达式。下面是一个简单的例子，展示如何绘制一个多段线（Piecewise Function）：

import matplotlib.pyplot as plt
import numpy as np

# 定义分段函数
def piecewise_function(x):
    if x < 0:
        return -x ** 2 + 4*x   # 第一段
    elif 0 <= x < 2:
        return x                   # 第二段
    else:
        return 2 * (x - 2)**2      # 第三段

# 设置x轴范围
x_values = np.linspace(-5, 6, 400)

# 计算对应y值
y_values = [piecewise_function(val) for val in x_values]

# 绘制图形
plt.figure(figsize=(8, 6))
plt.plot(x_values, y_values, label='分段函数')
plt.title('分段函数示例')
plt.xlabel('X轴')
plt.ylabel('Y轴')
plt.legend()
plt.grid(True)
plt.show()