Merge branch 'master' into main_dev

CoffeaTeam · Jun 2, 2023 · d156275 · d156275
2 parents 7d4e9c6 + 98f7874
commit d156275
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/coffea/nanoevents/methods/vector.py b/coffea/nanoevents/methods/vector.py
@@ -626,6 +626,12 @@ def pvec(self):
             behavior=self.behavior,
         )
 
+    @property
+    def rapidity(self):
+        pz = self.z
+        e = self.energy
+        return 0.5 * (numpy.log(e + pz) - numpy.log(e - pz))
+
     @property
     def boostvec(self):
         """The `x`, `y` and `z` components divided by `t` as a `ThreeVector`