#!/usr/bin/env python3
import pandas as pd
import numpy as np

df = pd.read_table("2022/counts/gene_name_TPM.tsv", index_col=0, usecols=["gene", "MDL1-CU0", "MDL2-CU1", "MDL5-LU0"])  # , "MDL6-LU1"
df['controlMeanTPM'] = df[["MDL1-CU0", "MDL5-LU0"]].mean(axis=1)
df['treatedMeanTPM'] = df[["MDL2-CU1"]].mean(axis=1)  # ,"MDL6-LU1"
df["log2FoldChange"] = np.log2((df["treatedMeanTPM"]+1e-6)/(df["controlMeanTPM"]+1e-6))

df.to_csv("2022/counts/1av2_treated_control_l2f.tsv", sep="\t")