在python中根据条件绘制多色线，但以零碎的线结束

import matplotlib.pyplot as plt

# 数据准备
x = [1, 2, 3, 4, 5]
y = [5, 3, 2, 4, 1]
conditions = [True, False, True, False, True]

# 绘制多色线
fig, ax = plt.subplots()
prev_condition = conditions[0]
prev_index = 0

for i in range(1, len(conditions)):
    if conditions[i] != prev_condition:
        if prev_condition:
            ax.plot(x[prev_index:i], y[prev_index:i], color='r')
        else:
            ax.plot(x[prev_index:i], y[prev_index:i], color='b')
        prev_condition = conditions[i]
        prev_index = i

# 处理最后一段线
if prev_condition:
    ax.plot(x[prev_index:], y[prev_index:], color='r')
else:
    ax.plot(x[prev_index:], y[prev_index:], color='b')

# 显示图形
plt.show()