Python - контур плотности графика с двумя различными размерными массивами

Question

Я пытаюсь сделать hexbin с двумя различными размерными массивами.

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$]') 

Это дает мне хороший участок как таковой.

То, что я хочу сделать его создать 2D-контур плотности от этого участка. Что-то подобное произведено здесь: https://micropore.wordpress.com/2011/10/01/2d-density-plot-or-2d-histogram/

Проблема в том, что у меня разные размерные массивы. Массив lam имеет форму (121209,3), а mass_msun имеет форму (121209,).

Есть ли способ сохранить размерные аспекты lam с помощью mass_sun? Или мне нужно использовать другую команду other histogram2d?

Спасибо!

Answer 1

Самый простой способ по-прежнему, чтобы изменить свои данные, чтобы соответствовать требованиям параметров Гистограмма:

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