w3resource

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


19. Pivot Table: Survival Probability by Class, Gender, Solo Boarding, and Port

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]

For more Practice: Solve these Related Problems:

  • Write a Pandas program to create a pivot table from titanic.csv that computes survival probability segmented by class, gender, solo boarding, and port of embarkation.
  • Write a Pandas program to generate a multi-dimensional pivot table that aggregates survival rates for combinations of class, gender, boarding status, and embarkation port.
  • Write a Pandas program to build a pivot table that cross-tabulates survival outcomes by class, gender, solo boarding status, and port, then displays the probabilities.
  • Write a Pandas program to create a pivot table with four dimensions (class, gender, solo boarding, port) and verify the computed survival percentages for each group.

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.

What is the difficulty level of this exercise?

Test your Programming skills with w3resource's quiz.



Follow us on Facebook and Twitter for latest update.