假设我们有一个数据集df,其中有一个列名为group,我们想要按组查找另一个列value中每个值的第一个出现位置。可以使用以下代码实现:
import pandas as pd
df = pd.DataFrame({'group': ['A', 'A', 'B', 'B', 'B', 'C', 'C'],
'value': [1, 2, 1, 3, 4, 2, 4]})
df['first_occurrence'] = df.groupby('value')['group'].transform('first')
这将在DataFrame中添加一列名为first_occurrence,其中包含列value中每个值的第一个出现位置所在的组。
输出结果为:
group value first_occurrence
0 A 1 A
1 A 2 A
2 B 1 A
3 B 3 B
4 B 4 B
5 C 2 A
6 C 4 B
下一篇:按组找到最大的行