用二维数组绘制密度等值线的Python算法

Question

我试着用两个不同的维数组做一个六边形。你知道吗

mass_msun = subhalos['SubhaloMass'] * 1e10 / 0.704 # in units of M_sol     h^-1
lam = (subhalos['SubhaloSpin'])

plt.plot(mass_msun,lam)
plt.xscale('log')
plt.yscale('log')
plt.ylabel('$\lambda$')
plt.xlabel('Subhalo Mass [$M_\odot$]')

这给了我一个很好的情节。你知道吗

我要做的是从这个图中创建一个二维密度等值线。类似的东西在这里产生：https://micropore.wordpress.com/2011/10/01/2d-density-plot-or-2d-histogram/

问题是，我有不同的维度数组。lam数组是形状的（121209,3），而mass_msun是形状的（121209,3）。你知道吗

有没有办法用mass_sun保存lam的维度方面？或者我应该继续使用其他historogram2d命令吗？你知道吗

谢谢你！你知道吗

Answer 1

最简单的方法仍然是重塑数据以符合直方图参数要求：

n_col = lam.shape[1]
H, xedges, yedges = np.histogram2d(np.tile(mass_msum, n_col), lam.flatten(), your_range, your_bins)