Mercurial > repos > laurenmarazzi > netisce_test
annotate 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 |
rev | line source |
---|---|
1 | 1 #!/usr/bin/env python3 |
2 import os | |
3 import numpy as np | |
4 import pandas as pd | |
5 import itertools | |
6 import csv | |
7 import sys | |
8 | |
9 def main(): | |
10 numnodes=len(open(sys.argv[1]).read().strip().split('\n')[1:])# number of perturbations to generate | |
11 prefix = sys.argv[2] # prefix for column labels | |
12 options=sys.argv[3].split(',') | |
13 test=itertools.product(options, repeat=numnodes) | |
14 df1=pd.DataFrame(list(test)) | |
15 df1=df1.set_index(prefix+'_'+df1.index.astype(str)) | |
16 | |
17 df1.to_csv('init.txt',sep=' ', index=True,na_rep='NA') | |
18 main() |