groupby函数获取每个组中的最后一个值 - Python

groupby函数获取每个组中的最后一个值 - Python

在Python中，pandas库提供了一个非常方便的函数 - groupby函数，用于将数据根据指定的一列或多列分组，并对每个组进行各种操作。其中，nth函数可以用来获取每个组中的指定位置上的值。

但有时候我们需要获取的是每个组中的最后一个值，该怎么办呢？这里介绍一个小技巧，即将获取最后一个值的问题转换为获取倒数第二个值，再用head(1)函数进行筛选。

下面是一个简单的示例代码：

import pandas as pd

# 创建示例数据
data = {'group': ['A', 'A', 'B', 'B', 'B', 'C'],
        'value': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)

# 获取每个组的最后一个值
last_vals = df.groupby('group').nth(-2).reset_index(drop=True)
last_vals = last_vals.groupby('group').head(1).reset_index(drop=True)
print(last_vals)

运行结果如下：

  group  value
0     A      1
1     B      4
2     C      6

在上面的代码中，我们首先用groupby和nth函数获取了每个组的倒数第二个值，然后再用groupby和head(1)函数获取了每个组的最后一个值，并将数据重新索引后输出。

通过这种方法，我们可以方便地获取每个组的最后一个值，并且不需要额外写循环等复杂的代码。