Source code for pybtls.output.Read.POT_counter

import pandas as pd
from pathlib import Path

__all__ = ["read_POT_C"]




def read_POT_C(
    file_path: Path, no_lines: int = None, start_line: int = 1
) -> pd.DataFrame:
    """
    Read the POT counter data from pybtls results.\n
    This output file has a header.

    Parameters
    ----------
    file_path : Path\n
        The path to the POT counter data file.\n
    no_lines : int, optional\n
        The number of data lines to read from the file.\n
        If not specified, all lines will be read.\n
    start_line : int, optional\n
        Default is 1. \n
        The line to start reading data from.

    Returns
    -------
    pd.DataFrame\n
        The POT counter data.
    """

    # Read data
    return_data = pd.read_csv(
        file_path,
        sep="[\s\t]+",
        header=None,
        skiprows=max(1, start_line),
        nrows=no_lines,
        engine="python",
    )
    no_effects = len(return_data.columns) - 1

    # Set column ids
    column_ids = ["Block"] + [f"Effect {i + 1}" for i in range(no_effects)]
    return_data.columns = column_ids

    return return_data