用matplotlib绘制不同颜色的y=0和y=1散点图

import matplotlib.pyplot as plt
import numpy as np

x = np.random.rand(100)  # 生成100个随机数作为x坐标
y = np.random.randint(0, 2, 100)  # 生成100个0和1之间的随机整数作为y坐标

x0 = []
y0 = []
x1 = []
y1 = []

for i in range(len(y)):
    if y[i] == 0:
        x0.append(x[i])
        y0.append(y[i])
    else:
        x1.append(x[i])
        y1.append(y[i])

plt.scatter(x0, y0, c='red', label='y=0')
plt.scatter(x1, y1, c='blue', label='y=1')

plt.legend()
plt.xlabel('x')
plt.ylabel('y')
plt.title('Scatter Plot of y=0 and y=1')

plt.show()