Mercurial > repos > laurenmarazzi > netisce_test
diff tools/myTools/bin/generate_perts.py @ 1:7e5c71b2e71f draft default tip
Uploaded
author | laurenmarazzi |
---|---|
date | Wed, 22 Dec 2021 16:00:34 +0000 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tools/myTools/bin/generate_perts.py Wed Dec 22 16:00:34 2021 +0000 @@ -0,0 +1,18 @@ +#!/usr/bin/env python3 +import os +import numpy as np +import pandas as pd +import itertools +import csv +import sys + +def main(): + numnodes=len(open(sys.argv[1]).read().strip().split('\n')[1:])# number of perturbations to generate + prefix = sys.argv[2] # prefix for column labels + options=sys.argv[3].split(',') + test=itertools.product(options, repeat=numnodes) + df1=pd.DataFrame(list(test)) + df1=df1.set_index(prefix+'_'+df1.index.astype(str)) + + df1.to_csv('init.txt',sep=' ', index=True,na_rep='NA') +main()