Pandas Pivot Titanic: Find the probability of survival by class, gender, solo boarding and port of embarkation

Last update on December 21 2024 07:52:27 (UTC/GMT +8 hours)

Write a Pandas program to create a Pivot table and find the probability of survival by class, gender, solo boarding and port of embarkation. Go to Editor

Sample Solution:

Python Code :

import pandas as pd
import numpy as np
df = pd.read_csv('titanic.csv')
result = df.pivot_table('survived', ['sex' , 'alone' ], [ 'embark_town', 'class' ])
print(result)

Sample Output:

embark_town  Cherbourg              ...    Southampton          
class            First    Second    ...         Second     Third
sex    alone                        ...                         
female False  1.000000  1.000000    ...       0.923077  0.327586
       True   0.944444  1.000000    ...       0.892857  0.466667
male   False  0.473684  0.166667    ...       0.300000  0.142857
       True   0.347826  0.250000    ...       0.089552  0.123762

[4 rows x 9 columns]

Python Code Editor:

Pivot Titanic.csv:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a Pandas program to create a Pivot table and separate the gender according to whether they traveled alone or not to get the probability of survival.