import uproot
import awkward as ak
from coffea.nanoevents import NanoEventsFactory, NanoAODSchema
from coffea import processor, hist
import os, shutil
if not os.path.isfile("nano_dy.parquet"):
ak.to_parquet(
uproot.lazy("nano_dy.root:Events"),
"nano_dy.parquet",
list_to32=True,
use_dictionary=False,
compression="GZIP",
compression_level=1,
)
if not os.path.isfile("nano_dimuon.parquet"):
ak.to_parquet(
uproot.lazy("nano_dimuon.root:Events"),
"nano_dimuon.parquet",
list_to32=True,
use_dictionary=False,
compression="GZIP",
compression_level=1,
)
if not os.path.isdir("/mnt/cephfs/nanoevents/ZJets"):
os.makedirs("/mnt/cephfs/nanoevents/ZJets")
for i in range(2):
shutil.copyfile('nano_dy.parquet', '/mnt/cephfs/nanoevents/ZJets/nano_dy.{}.parquet'.format(i))
if not os.path.isdir("/mnt/cephfs/nanoevents/Data"):
os.makedirs("/mnt/cephfs/nanoevents/Data")
for i in range(2):
shutil.copyfile('nano_dimuon.parquet', '/mnt/cephfs/nanoevents/Data/nano_dimuon.{}.parquet'.format(i))
events_skyhook = NanoEventsFactory.from_parquet("/mnt/cephfs/nanoevents/ZJets/nano_dy.0.parquet",
skyhook_options = {"ceph_config_path": "/opt/ceph/ceph.conf",
"ceph_data_pool": "cephfs-data0"}
).events()
ak.flatten([events_skyhook.Muon[i].pt for i in range(len(events_skyhook.Muon)) if len(events_skyhook.Muon[i])]).to_numpy()
/opt/conda/lib/python3.8/site-packages/coffea/nanoevents/schemas/nanoaod.py:193: RuntimeWarning: Missing cross-reference index for FatJet_genJetAK8Idx => GenJetAK8 warnings.warn(
array([76.75331879, 20.13140869, 31.03870392, 50.64134216, 14.33010292, 16.72498322, 13.90806293, 46.49888611, 40.66781235, 51.4357605 , 39.5927887 , 38.89543152, 33.71282196, 17.08279228, 14.52699471, 4.3605423 , 10.11770916, 17.94991875])