Functionality like root, SetLogz

[yszhang95]

Dear experts

I am a newbie of python. Forgive me if I make a stupid question.

I tried to play with 2d histograms with mplhep. I am trying to plot it like ROOT code

TH2D::Draw("COLZ");
TCanvas::SetLogz();

Do we have such functionality in this package or we have to rely on other packages?

Could you please give an example regarding this need?

I found we can use matplotlib.pyplot.yscal('log') to modify x and y axes. But I cannot find any stuff for z axis.

Many thanks in advance.

[andrzejnovak]

Hi @yszhang95 you can pass "norm" argument to histplot2d which gets passed on to the underlying method pcolormesh

import numpy as np
import matplotlib.pyplot as plt
import matplotlib
import mplhep as hep

x = np.linspace(0, 5, 10)
y = np.linspace(5, 10, 10)
H = np.random.uniform(0.01, 20, (9,9))

f, ax = plt.subplots()
hep.hist2dplot(H, x, y, norm=matplotlib.colors.LogNorm(vmin=0.1))

Generally a lot of matplotlib solutions can be found on stack overflow https://stackoverflow.com/questions/17201172/a-logarithmic-colorbar-in-matplotlib-scatter-plot