PLT绘图

1、绘制不同颜色的点（二维）

import matplotlib.pyplot as plt

if __name__ == '__main__':
    # 准备数据
    x = [1, 2, 3, 4, 5]  # X轴上的点
    y = [1, 4, 9, 16, 25]  # Y轴上的点，这里以x的平方为例
    colors = ['red', 'green', 'blue', 'yellow', 'purple']  # 点的颜色列表

    # 绘制点
    for i in range(len(x)):
        plt.scatter(x[i], y[i], color=colors[i])  # 使用scatter函数绘制点，并指定颜色

    # 添加标题和坐标轴标签
    plt.title('myTitle')
    plt.xlabel('X axis')
    plt.ylabel('Y axis')

    # 显示图形
    plt.show()

效果：

还可以绘制多个画布：

import matplotlib.pyplot as plt

if __name__ == '__main__':
    # 准备数据
    x = [1, 2, 3, 4, 5]  # X轴上的点
    y = [1, 4, 9, 16, 25]  # Y轴上的点，这里以x的平方为例
    colors = ['red', 'green', 'blue', 'yellow', 'purple']  # 点的颜色列表

    fig = plt.figure()  # 创建一个图形画布，可以增加子图形

    ax1 = fig.add_subplot(1, 2, 1)  # 1行2列，第1个位置
    ax1.set_title('Subplot 1')
    ax1.scatter(x,y, s=7, c='red')  # 绘制点；s表示点的大小

    ax2 = fig.add_subplot(1, 2, 2)  # 1行2列，第2个位置
    ax2.set_title('Subplot 2')
    ax2.scatter(x, y, s=1, c='green')  # 绘制点；s表示点的大小

    # 显示图形
    plt.show()