mirror of
https://github.com/vale981/bachelor_thesis
synced 2025-03-10 12:56:41 -04:00
18 lines
370 B
Python
18 lines
370 B
Python
"""This module defines some observables on arrays of 4-pulses."""
|
|
import numpy as np
|
|
|
|
def p_t(p):
|
|
"""Transverse momentum
|
|
|
|
:param p: array of 4-momenta
|
|
"""
|
|
|
|
return np.linalg.norm(p[:,1:3], axis=1)
|
|
|
|
def η(p):
|
|
"""Pseudo rapidity.
|
|
|
|
:param p: array of 4-momenta
|
|
"""
|
|
|
|
return np.arccosh(np.linalg.norm(p[:,1:], axis=1)/p_t(p))*np.sign(p[:, 3])
|