DataFrame操作巩固2——性别和等级对共同生还率的影响

重点练习 groupby() 函数的使用

1. 数据预处理：

import pandas as pd
from pandas import DataFrame, Series
import numpy as np

df = pd.read_csv('./exercise_data/train.csv')
df.head()

### 预处理
'''
# 将性别转化为数字，法1
def fun1(Sex):
    if Sex == 'male':
        return 1
    else:
        return 0
df['Sex'] = df['Sex'].apply(fun1)
df
'''

# 将性别转化为数字，法2
df['Sex'] = df['Sex'].map({'male':1, 'female':0})
df.head()

2. 性别和等级对共同生还率的影响, 利用groupby() 函数

# 按性别与等级对乘客进行分组
group_all = df.groupby(['Sex','Pclass']).count()['PassengerId']
group_all

 df.groupby(['Sex','Pclass']).count()：

按照性别和等级进行分组，并且计数，返回的是一个DataFrame，行索引为性别和等级对