python - 如何在 python 中从 3 个数组创建颜色映射

Question

我正在尝试在 python 中创建两个数组 t1 和 t2 的颜色图，颜色由第三个 v 设置，但我无法根据 v 数组获得颜色条，它而是根据 t1。这是我的代码:

    import matplotlib.pyplot as plt
    import numpy as np
    t1 = [75, 76, 77, 78]
    t2 = [75, 76, 77, 78]
    v = [0.5, 0.5, 0.8, 0.8]

    image_data = np.column_stack([t1, t2, v])
    plt.imshow(image_data)
    plt.colorbar()
    plt.show()

它产生这个数字:

如有任何帮助，我们将不胜感激。

Answer 1

您不能使用 imshow 设置 x 和 y 坐标，并将颜色设置为 3rd。就是显示一个矩阵图像，其中有X*Y值，所有这些值代表颜色。也许您想使用 scatter。例如。你可以试试:

import matplotlib.pyplot as plt
t1 = [0,1,2,3]
t2 = [0, 10, 20, 30]
v = [0.5, 0.5, 0.8, 0.8]

plt.scatter(t1, t2, c=v, cmap='Greens')
plt.colorbar()
plt.show()

您可以查看最适合您的颜色图。