In [18]:

    

import pandas as pd 


def readCSVfile(file):
    return pd.read_csv(file)
 
def readCDfile(file):
    return pd.read_csv(file,delimiter=":")

def determineFileType(df):
    if len(df.columns) == 4:
        alt = 1
    if len(df.columns) == 1:
        alt = 2
        df = readCDfile(file)
    if len(df.columns) == 7:
        alt = 3
    return alt, df

def parseFile(alt,df):
    if alt == 1:
        df['One'] = 'a'
        df['Two'] = 'b'
        df['Three'] = 'c'
        df['Four'] = 'd'
    if alt == 2:
        df["TWO"][df["ONE"] == 1] = 3
        df["TWO"][df["ONE"] == 2] = 4
        df["TWO"][df["ONE"] == 3] = 6
        df["ONE"][df["TWO"] == 3] = 1
        df["ONE"][df["TWO"] == 4] = 2
        df["ONE"][df["TWO"] == 6] = 3
    if alt == 3:
        df['One'] = 1
        df['Two'] = 2
        df['Three'] = 3
        df['Four'] = 4
        df['Five'] = 5
        df['Six'] = 6
        df['Seven'] = 7
    return df
    

def writeFile(df,alt):
    df.to_csv('DEST_AWW_alt' + str(alt) +'.csv',index = False, sep='\t')


    

In [21]:

    

file = input("Enter the name of the file: ")

df = readCSVfile(file)
alt, df = determineFileType(df)
df = parseFile(alt,df)
writeFile(df,alt)


    

    




Enter the name of the file: source.csv

In [ ]: