CODEWITHSUNDEEP
pythonpandasdataframetype-conversion
krakowi
krakowi

Reputation: 613

Convert string into float for multiple columns in data frame

I am playing with the data set from FIFA 19 and currently trying to convert string value, wage and release clauses into float values e.g. €110.5M into 110500000. Here is my function which takes text as an argument and returns float value:

def text_to_num(text):
    d = {'K': 3, 'M': 6, 'B': 9}
    new_text = text[1:]
    if new_text[-1] in d:
        num = new_text[:-1]
        magnitude = new_text[-1]
        return Decimal(num) * 10 ** d[magnitude]
    else:
        return Decimal(new_text)

It's quite easy to apply it on single column, e.g. Value:

def convert_into_float(data):
    data['Value'] = data['Value'].apply(text_to_num)
    return data

I am wondering, how to convert all 3 columns at once instead of using above function three times. 

def main():
    # load the data from .csv file, use ID column as index  
    new_data = load_data("data.csv", 'ID') 
    new_data = convert_into_float(new_data)

Upvotes: 3

Views: 996

Answers (1)

jezrael
jezrael

Reputation: 863741

You can use DataFrame.applymap with filtered columns by list:

def convert_into_float(data):
    cols = ['col1','col2','col3']
    data[cols] = data[cols].applymap(text_to_num)
    return data

Upvotes: 1

Related Questions