import pandas as pd
import servicex as sx
import awkward as ak
from func_adl_servicex import ServiceXSourceUpROOT
import matplotlib.pyplot as plt
plt.rcParams['figure.dpi'] = 200 # make figures bigger and more readable
dataset_name = ["root://eospublic.cern.ch//eos/root-eos/benchmark/Run2012B_SingleMu.root"]
sx_dataset = sx.ServiceXDataset(dataset_name, "uproot")
ds = ServiceXSourceUpROOT(sx_dataset, "Events")
jet_pT = ds.Select(lambda event: event.Jet_pt).AsAwkwardArray().value()
ak.flatten(jet_pT)
<Array [71.7, 27.8, 23.2, ... 19.9, 15.3, 17] type='170952895 * float32'>
plt.hist([ak.flatten(jet_pT)], bins=100, range=(0, 100))
plt.xlabel(r'$p_\mathrm{T}$ [GeV]')
plt.ylabel('Jets')
plt.show()