来自多维数组的热图

Question

我试图在python中从两个多维数组（199,199）生成热图。执行该函数时，这是我得到的错误：

文件＆＃34; /home/jorge/Escritorio/Jorge/scripts/energy_landscape/energy_landscape.py"，第265行，在heat_map中     Xi = np.r_ [rvs [：，0] .min（）：rvs [：，0] .max（）：199j] IndexError：索引太多

我想这是因为数据庞大。我怎么处理它？ 这是我使用的代码：

def heat_map(RMSD, E):
    rvs = np.append(RMSD, E)
    kde = stats.kde.gaussian_kde(rvs.T)

    Xi = np.r_[rvs[:,0].min():rvs[:,0].max():199j]
    Yi = np.r_[rvs[:,1].min():rvs[:,0].max():199j]
    X,Y = np.meshgrid(xi, yi)
    grid_coords = np.append(X.reshape(-1,1), Y.reshape(-1,1), axis = 1)

    Z = kde(grid_coords.T)
    Z = z.reshape(199,199)

    scatter(rvs[:,0], rvs[:,1], alpha=1 ,color='white')
    plot = imshow(Z, aspect=Xi.ptp()/Yi.ptp(), origin='lower', extent=(rvs[:,0].min(), rvs[:,0].max(), rvs[:,1].min(), rvs[:,1].max()))

    return plot

谢谢！ ;）

注意：RMSD和E是（199,199）python列表