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

Uploaded

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