matplotlib绘图矩阵保持原始坐标

Question

我有两个向量x和y以及一个矩阵，z[i,j] = f(x[i], y[j])

我想在z和x上标注y在坐标轴上的坐标。换句话说，点z[i, j]应保留在图中的x[i], y[j]位置。

我需要保持这种关系，因为我需要重叠不同的情节。

plt.imshow(z)不起作用，z[i,j]位于(i, j)而不是(x[i], x[j])

我该如何解决？

编辑： 我需要一个二维表示

Answer 1

你可能会使用散点图。

import matplotlib.pyplot as plt
import numpy as np
N = 40
X = np.random.rand(N)
Y = np.random.rand(N)
def f(x,y): #function f
    return x+y
Z = np.zeros([N,N])
for i in range(N):
    for j in range(N):
        Z[i,j] = f(X[i],Y[j])
        plt.scatter(X[i],Y[j],c=Z[i,j],s=40,vmin=0.0,vmax=2.0,cmap='viridis',marker='s')

plt.show()