Python 多维数组的热图

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

文件“/home/jorge/Escritorio/jorge/scripts/energy_-scape/energy_-scape.py”，第265行，热图 席= NP .r[rvs[，0 ] min（）：rvs[，0 ] .（上）： 索引器：索引太多

我想这是因为数据量很大。我该怎么处理呢？ 下面是我使用的代码：

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是（199199）python列表