diff tools/myTools/bin/generate_perts.py @ 1:7e5c71b2e71f draft default tip

Uploaded

--- /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()