bachelor_thesis/prog/python/qqgg/tangled/observables.py

"""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-momentums
    """

    return np.linalg.norm(p[:,1:3], axis=1)

def η(p):
    """Pseudo rapidity.

    :param p: array of 4-momentums
    """

    return np.arccosh(np.linalg.norm(p[:,1:], axis=1)/p_t(p))*np.sign(p[:, 3])