圖層影像和實時更新熱圖

我有：

• 帶有 16 個壓力傳感器的鞋墊的影像。
• 每個傳感器的 2*8 numpy 壓力值陣列
• 壓力值每 0.1[s] 實時更新一次

我想要：

• 在繪圖背景中顯示影像
• 疊加到每個壓力傳感器（黑點）施加壓力的熱圖值
• 每 0.1[s] 更新一次壓力值

下面的代碼確實用背景影像覆寫了更新熱圖，但是我不知道如何重新定位熱圖以覆寫傳感器（黑點）

import matplotlib.pyplot as plt
import numpy as np
import time
 #Trial combining heatmap   backgoround image
fig = plt.figure()
ax = fig.add_subplot(111)
im = ax.imshow(np.random.random((300,400))) # Usually 2*8 but cannot see it 
img = plt.imread(r"..\Desktop\Example_image.PNG")

ax.imshow(img, alpha=0.5)
plt.show(block=False)

# draw some data in loop
for i in range(100):
    # wait for a second
    time.sleep(0.1)
    # PRESSURE SENSOR VALUES USUALLY 2*8, USING 300*400 SO THEY ARE VISIBLE
    Pressue_values = np.random.random((300,400))
    im.set_array(Pressue_values)
    # redraw the figure
    fig.canvas.draw()
    fig.canvas.flush_events()
    
    

如何將 2*8 壓力值映射到正確的壓力傳感器像素？

當前結果示例 - 像素太小，需要映射到正確位置并擴展到相鄰

uj5u.com熱心網友回復：

此代碼使用散點圖在壓力傳感器上繪制圓圈并根據輸入資料更新顏色。您應該更新SENSOR_MIN并SENSOR_MAX匹配您的傳感器（這可以防止色標在輸入發生變化時發生變化）。

import sys
import time
import matplotlib.pyplot as plt
import numpy as np

# min and max values that the sensors will produce
SENSOR_MIN = 0
SENSOR_MAX = 1

def make_plot(background, coordinates, pressure_data):
    fig, ax = plt.subplots(1, 1)
    ax.imshow(background)

    coordinates = np.asarray(coordinates).reshape(-1, 2).T
    sc = ax.scatter(
        *coordinates,
        s=10.0**2,
        c=pressure_data.reshape(-1),
        cmap="viridis",
        vmin=SENSOR_MIN,
        vmax=SENSOR_MAX,
    )
    fig.colorbar(sc, label="pressure (some units)")
    return fig, sc

def update_plot(fig, sc, pressure_data):
    """Update the plot with new data and redraw the figure."""
    sc.set_array(pressure_data.reshape(-1))
    fig.canvas.draw()
    fig.canvas.flush_events()

def get_data():
    return np.random.random((2, 8))

if __name__ == "__main__":
    background_img = plt.imread("Example_image.PNG")
    # pixel coordinates of each sensor in the background image
    coords = [[[462, 106],
               [547, 113],
               [430, 156],
               [601, 237],
               [543, 222],
               [497, 221],
               [430, 216],
               [444, 266]],
              [[489, 439],
               [609, 463],
               [613, 564],
               [500, 574],
               [568, 641],
               [524, 643],
               [577, 688],
               [527, 683]]]

    fig, sc = make_plot(background_img, coords, get_data())
    # exit the script when the figure is closed
    fig.canvas.mpl_connect("close_event", lambda event: sys.exit())
    plt.show(block=False)

    # draw some data in loop
    while True:
        frame_start = time.perf_counter()
        pressure_values = get_data()
        # update the figure
        update_plot(fig, sc, pressure_values)
        # wait for 0.1s (including the time it took to update the plot)
        frame_time = time.perf_counter() - frame_start
        if frame_time < 0.1:
            plt.pause(0.1 - frame_time)

標籤：Python 图片 麻木的 matplotlib 热图

上一篇：如何允許影像在SwiftUI中超出Form{}？

下一篇：嘗試通過id獲取youtube中的特定視頻