from colossus.lss import mass_function

​

z = [0.0, 1.0, 2.0, 4.0]

M = 10**np.arange(11.0, 15.5, 0.1)

​

plt.figure()

plt.xlabel('M200m')

plt.ylabel('dn/dln(M)')

plt.loglog()

plt.xlim(1E11, 4E15)

plt.ylim(1E-7, 1E-1)

for i in range(len(z)):

    mfunc = mass_function.massFunction(M, z[i], mdef = '200m', model = 'tinker08', q_out = 'dndlnM')

    plt.plot(M, mfunc, '-', label = 'z = %.1f' % (z[i]))

plt.legend();