Numpy Array to LaTeX Table

numpy_to_latex_table.py

import numpy as np
import pandas as pd

def numpy_to_latex(
  arr: np.ndarray,
  columns=None,
  index=False,
  caption=None,
  label=None,
  float_format="%.3f",
  **to_latex_kwargs
):
    df = pd.DataFrame(arr, columns=columns)
    latex_kwargs = {
        "index": index,
        "caption": caption,
        "label": label,
        "float_format": float_format,
        **to_latex_kwargs
    }
    for key in ("caption", "label"):
        if latex_kwargs.get(key) is None:
            del latex_kwargs[key]
    return df.to_latex(**latex_kwargs)

if __name__ == "__main__":
  np.random.seed(0)
  arr = np.random.randn(4, 3)
  cols = ["Feature 1", "Feature 2", "Feature 3"]
  latex_code = numpy_to_latex(
    arr,
    columns=cols,
    caption="My random data",
    label="tab:random",
    column_format="lrrr",
    longtable=True
  )
  print(latex_code)