atompy.load_2d_from_root#
- atompy.load_2d_from_root(fname, hname, output_format='pcolormesh')[source]#
Load 2D data stored in a ROOT file.
- Parameters:
- Returns:
- output
PcolormeshData,ImshowDataorHist2d Depends on output_format.
output_format == "pcolormesh": ReturnPcolormeshData.output_format == "imshow": ReturnImshowData.output_format == "Hist2d": Return aHist2d.
- output
Examples
Load data such that it can be plotted using
matplotlib.pyplot.pcolormesh().data = ap.load_2d_from_txt("rootfile.root", "path/to/hist", output_format="pcolormesh") plt.pcolormesh(data.x, data.y, data.z)
Load data such that it can be plotted using
matplotlib.pyplot.imshow().data = ap.load_2d_from_txt("rootfile.root", "path/to/hist", output_format="imshow") plt.imshow(data.image, extent=data.extent)
Alternatively, immediately unpack the loaded data into their respective
numpy.ndarray.x, y, z = ap.load_2d_from_txt("rootfile.root", "path/to/hist", output_format="pcolormesh") plt.pcolormesh(x, y, z) image, extent = ap.load_2d_from_txt("rootfile.root", "path/to/hist", output_format="imshow") plt.imshow(image, extent=extent)
Load data as a
Hist2d".data = ap.load_2d_from_txt("rootfile.root", "path/to/hist", output_format="Hist2d") data = hist.column_normalized_to_sum.for_pcolormesh plt.pcolormesh(data.x, data.y, data.z)