plot_coexpression_error_by_sparsity.py

#!/usr/bin/env python

### Imports ###
import pandas as pd
import numpy as np

from pathlib import Path
from typing import Iterable

from lib import plot_boxplots

### Functions ###
def plot_coexpression_error_by_sparsity(
    input_files: Iterable[Path],
    sparsity_levels: Iterable[float],
    output: Path,
) -> None:
    """
    Plots the statistics for the coexpression error from the
    provided files as a boxplot.

    Parameters
    ----------
    input_files : Iterable[Path]
        Iterable of Paths to files containing the data
    sparsity_levels : Iterable[float]
        Iterable of sparsity levels corresponding to the files
    output : Path
        Path to the file where the plot will be saved
    """

    df = pd.DataFrame()
    for sparsity,file in zip(sparsity_levels, input_files):
        df[sparsity] = np.load(file)
        df[sparsity] = df[sparsity].fillna(df[sparsity].mean())

    fig,_ = plot_boxplots(df, corr_name=None, ylim=(0, 2))
    fig.savefig(output, dpi=300, bbox_inches='tight')

### Main body ###
if __name__ == '__main__':
    from argparse import ArgumentParser

    parser = ArgumentParser()
    parser.add_argument('-i', '--input', dest='input_files', nargs='+',
        help='files to plot the data from', metavar='FILE')
    parser.add_argument('-sl', '--sparsity_levels', dest='s_levels', nargs='+',
        help='levels of sparsity corresponding to the files', metavar='FLOAT',
        type=float)
    parser.add_argument('-o', '--output', dest='output',
        help='file to save the plot to', metavar='FILE')

    args = parser.parse_args()

    plot_coexpression_error_by_sparsity(
        args.input_files,
        args.s_levels,
        args.output,
    )