PLT繪圖

东方春發表於2024-08-23
繪圖

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()

相關文章