In [155]:
import pandas as pd
import re
import numpy as np
In [164]:
choice = 'alt2.csv'
m,n = (pd.read_csv(choice).shape)
if m == 7 and n ==4:
df1_out = pd.DataFrame([list('abcd')], index=range(7))
df1_out.columns = ['One', 'Two','Three','Four']
df1_out.to_csv('Dest_CZ_atl1.tsv',index=False,sep='\t')
elif m == 9 and n == 1:
df2_l = list(df2['ONE:TWO'])
df2_l2 = []
for i in df2_l:
df2_l2.append((re.split(r'\:', i)))
value_1 = []
value_2 = []
for i in df2_l2:
value_1.append((i[0]))
value_2.append((i[1]))
df2_final = pd.DataFrame({'One':value_1,'Two': value_2})
df2_tuple = []
for i,j in zip(value_1,value_2):
df2_tuple.append((i,j))
arr = np.asarray(df2_final)
for i in arr:
if i[0] == '1':
i[1] = 3
elif i[0] == '2':
i[1] = 4
elif i[0] == '3':
i[1] = 6
elif i[1] == '3':
i[0] = 1
elif i[1] == '4':
i[0] = 2
elif i[1] == '6':
i[0] = 3
df2_final = pd.DataFrame(arr)
df2_final = df2_final.rename(columns={0:'One', 1:'Two'})
df2_final.to_csv('Dest_CZ_atl2.tsv',index=False,sep='\t')
elif m == 3 and n == 7:
df3_out = pd.DataFrame([list('1234567')], index=range(3))
df3_out.columns = ['One', 'Two','Three','Four','Five','Six','Seven']
df3_out.to_csv('Dest_CZ_atl3.tsv',index=False,sep='\t')