import pandas as pd
import awkward as ak
import servicex as sx
from func_adl_servicex import ServiceXSourceUpROOT
import matplotlib.pyplot as plt
plt.rcParams['figure.dpi'] = 200 # make figures bigger and more readable
dataset_name = "cms:DYJetsToLL_M-50_TuneCP5_13TeV-amcatnloFXFX-pythia8/RunIIAutumn18NanoAODv7-Nano02Apr2020_102X_upgrade2018_realistic_v21_ext2-v1/NANOAODSIM"
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 [156, 104, 50.5, ... 17.8, 17.2, 16.5] type='976406625 * float32'>
plt.hist([ak.flatten(jet_pT)], bins=100, range=(0, 100))
plt.xlabel(r'$p_\mathrm{T}$ [GeV]')
plt.ylabel('Jets')
plt.show()