在python中使用if-else条件语句绘制

if condition:
    # 如果条件为真，则执行这里的代码
else:
    # 如果条件为假，则执行这里的代码

import matplotlib.pyplot as plt
import numpy as np

# 生成一些示例数据
x = np.linspace(0, 10, 100)
y = np.sin(x)

# 创建一个新的图形
plt.figure()

# 使用if-else条件语句根据y的值绘制不同的颜色
for i in range(len(x)):
    if y[i] > 0:
        plt.plot(x[i], y[i], 'go')  # 如果y[i]大于0，则绘制绿色圆点
    else:
        plt.plot(x[i], y[i], 'ro')  # 如果y[i]小于或等于0，则绘制红色圆点

# 添加标题和坐标轴标签
plt.title('Sine Wave with Conditional Coloring')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')

# 显示图形
plt.show()

import matplotlib.pyplot as plt
import numpy as np

# 生成一些示例数据
x = np.linspace(0, 10, 100)
y = np.sin(x)

# 创建一个新的图形
plt.figure()

# 使用条件索引直接绘制不同颜色的点
plt.plot(x[y > 0], y[y > 0], 'go')  # 绘制y大于0的绿色圆点
plt.plot(x[y <= 0], y[y <= 0], 'ro')  # 绘制y小于或等于0的红色圆点

# 添加标题和坐标轴标签
plt.title('Sine Wave with Conditional Coloring')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')

# 显示图形
plt.show()