awakward_copy.py

awakward_copy.py

import awkward as ak

events = ak.Array([
    {"Muon": [{"pt": 10.0}, {"pt": 20.0}]},
    {"Muon": [{"pt": 30.0}]},
])

print("Original pt:")
print(events.Muon.pt)

# Save pt_raw from pt
events["Muon", "pt_raw"] = events.Muon.pt

print("\nAfter pt_raw = pt:")
print("pt     :", events.Muon.pt)
print("pt_raw :", events.Muon.pt_raw)

# Replace pt with a new array
events["Muon", "pt"] = events.Muon.pt + 100

print("\nAfter replacing pt:")
print("pt     :", events.Muon.pt)
print("pt_raw :", events.Muon.pt_raw)