使用.groupby().transform进行数据变换

使用 pandas 编写高效代码

Leonidas Souliotis

PhD Candidate

餐厅数据集

| total_bill   | tip   | sex      | smoker   | day   | time     | 
|--------------|-------|----------|----------|-------|----------|
| 16.99        | 1.01  | Female   | No       | Sun   | "Dinner" | 
| 10.34        | 1.66  | Male     | No       | Sun   | "Dinner" |

restaurant_grouped = restaurant.groupby('smoker')

print(restaurant_grouped.count())

|                  | total_bill |   tip |   sex  |  day   |   time |
|------------------|------------|-------|--------|--------|--------|
| smoker           |            |       |        |        |        |
| No               | 151        | 151   | 151    | 151    | 151    |
| Yes              | 93         | 93    | 93     | 93     | 93     |

数据变换

zscore = lambda x: (x - x.mean()  ) / x.std()

restaurant_grouped = restaurant.groupby('time')
restaurant_transformed = restaurant_grouped.transform(zscore)

restaurant_transformed.head()

   total_bill       tip      size
0   -0.416446 -1.457045 -0.692873
1   -1.143855 -1.004475  0.405737
2    0.023282  0.276645  0.405737
3    0.315339  0.144355 -0.692873
4    0.414880  0.353234  1.504347

与原生方法对比

restaurant.groupby('sex').transform(zscore)

mean_female = restaurant.groupby('sex').mean()['total_bill']['Female']
mean_male = restaurant.groupby('sex').mean()['total_bill']['Male']
std_female = restaurant.groupby('sex').std()['total_bill']['Female']
std_male = restaurant.groupby('sex').std()['total_bill']['Male']

for i in range(len(restaurant)):
    if restaurant.iloc[i][2] == 'Female':
        restaurant.iloc[i][0] = (restaurant.iloc[i][0] - mean_female)/std_female
    else:
        restaurant.iloc[i][0] = (restaurant.iloc[i][0] - mean_male)/std_male

使用.groupby()的耗时：0.016291141 秒
使用原生 Python 的耗时：3.937326908 秒

时间差：24,068.5145%

让我们来练习！

使用 pandas 编写高效代码