CODEWITHSUNDEEP
pythonpandasdataframe
musician1023
musician1023

Reputation: 55

How to convert multiple pandas columns from string boolean to boolean?

I have this dataframe

data = [['Tom', 16, 'True','False'], ['Nick', 19, 'False','True'], ['Juli', 17, 'True','True']] 
df = pd.DataFrame(data, columns = ['Name', 'Age', 'Writer','Artist'])

I want to convert the string booleans to booleans.

I have tried

def str_to_bool(s):
    if s == 'True':
         return True
    elif s == 'False':
         return False
    else:
         raise ValueError

df[['Writer','Artist']] = df[['Writer','Artist']].apply(str_to_bool)

and

import ast
df[['Writer','Artist']] = df['Writer','Artist'].map(ast.literal_eval)

but neither of these worked.

Is it possible to do convert the type of multiple columns in a single line or do I have to convert the relevant columns one at a time?

Upvotes: 0

Views: 1307

Answers (2)

Jason Cook
Jason Cook

Reputation: 1501

You can use the dictionary version of replace.

df = df.replace({'True': True, 'False': False})

df
   Name  Age  Writer  Artist
0   Tom   16    True   False
1  Nick   19   False    True
2  Juli   17    True    True

dtypes
Name      object
Age        int64
Writer      bool
Artist      bool
dtype: object

Upvotes: 2

Derek Eden
Derek Eden

Reputation: 4618

you could just do:

df['Writer'].apply(eval)

output:

>>> df['Writer'].apply(eval)
0     True
1    False
2     True
Name: Writer, dtype: bool

Upvotes: -1

Related Questions