使用Python繪制實時的動態(tài)折線圖

更新時間：2024年11月18日 09:10:59 作者：boss-dog

這篇文章主要為大家詳細介紹了如何使用Python繪制實時的動態(tài)折線圖,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下

1.數(shù)據(jù)繪制

def draw_data():
    index = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    x_data = [1, 0.2, 0.3, 4, 0.5, 0.6, 1, 0.8, 0.9, -1]

    # 創(chuàng)建折線圖
    plt.plot(index, x_data, marker='o', color='b', linestyle='-', label='x_data')
    # 設置標題和標簽
    plt.title("x_data")
    plt.xlabel("Index")
    plt.ylabel("X Data")
    # 顯示圖例
    plt.legend()
    # 設置橫坐標刻度，使得每個index值都顯示
    plt.xticks(index)
    # 顯示圖形
    plt.show()

2.繪制實時的動態(tài)折線圖

雖然可以實時繪制，但會不斷新增新的窗口，導致越到后面越卡頓，后面采用了保存到CSV文件進行分析的方法。

def realtime_data_draw():
    '''
    動態(tài)折線圖實時繪制
    '''
    plt.ion()
    plt.figure(1)
    t_list = []
    result_list = []
    t = 0

    while True:
        if t >= 100 * np.pi:
            plt.clf()
            t = 0
            t_list.clear()
            result_list.clear()
        else:
            t += np.pi / 4
            t_list.append(t)
            result_list.append(np.sin(t))
            plt.plot(t_list, result_list, c='r', ls='-', marker='o', mec='b', mfc='w')  ## 保存歷史數(shù)據(jù)
            plt.plot(t, np.sin(t), 'o')
            plt.pause(0.1)

3.保存實時數(shù)據(jù)到CSV文件中

將實時的數(shù)據(jù)保存到CSV文件中，通過excel文件繪制折線圖進行分析。

def realtime_data_save_csv():
    # 模擬實時生成的軌跡點坐標
    count = 0

    # CSV 文件路徑
    file_path = 'vision_data/pose.csv'
    if os.path.exists(file_path):
        os.remove(file_path)

    # 寫入表頭并開始寫入數(shù)據(jù)
    with open(file_path, mode='w', newline='') as file:
        writer = csv.writer(file)
        # 寫入表頭
        writer.writerow(['Index', 'X', 'Y', 'Z', 'RX', 'RY', 'RZ'])

        while True:
            count += 1
            x_value = random.uniform(-0.5, 0.5)
            y_value = random.uniform(-0.5, 0.5)
            z_value = random.uniform(-0.1, 0.8)
            rx_value = random.uniform(-3.14, 3.14)
            ry_value = random.uniform(-3.14, 3.14)
            rz_value = random.uniform(-3.14, 3.14)
            # 將生成的數(shù)據(jù)寫入 CSV 文件
            writer.writerow([count, x_value, y_value, z_value, rx_value, ry_value, rz_value])
            time.sleep(0.05)