Minor performance and readability changes.

[ajinkya-kulkarni]

Thanks for this repo! I came across this in the bioRxiv preprint. I made a few changes in the code:

1. Added a docstring to describe the function's purpose, arguments, and return value.
2. Added comments to explain the code logic and any complex computations.
3. Used more descriptive variable names, for example, num_bins instead of bins.
4. Added error checking for invalid input arguments or unexpected data shapes.
5. Used consistent naming conventions throughout the code, such as using underscores to separate words in variable names.
6. Separated the code into smaller functions with specific responsibilities to improve modularity and readability.
7. Used NumPy functions to simplify some of the computations, for example, np.linalg.norm() instead of np.sqrt(x2 + y2).

[ChiaraStringari]

Hi Ajinkya,
Thanks for your proposition and pull request.
We will get back soon to the code optimization and we will look also at this.
Best,
Chiara