下面是一个示例代码,可以按Column1进行分组,然后获取Column2上第一个和最后一个元素的所有元素:
import pandas as pd
# 创建示例数据框
data = {'Column1': ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
'Column2': [1, 2, 3, 4, 5, 6, 7, 8, 9]}
df = pd.DataFrame(data)
# 按Column1进行分组,并获取Column2上第一个元素的所有元素
first_elements = df.groupby('Column1')['Column2'].first().reset_index()
# 按Column1进行分组,并获取Column2上最后一个元素的所有元素
last_elements = df.groupby('Column1')['Column2'].last().reset_index()
print("第一个元素的所有元素:")
print(first_elements)
print("\n最后一个元素的所有元素:")
print(last_elements)
输出结果如下:
第一个元素的所有元素:
Column1 Column2
0 A 1
1 B 4
2 C 7
最后一个元素的所有元素:
Column1 Column2
0 A 3
1 B 6
2 C 9
在示例代码中,我们使用了groupby
函数按Column1进行分组。然后,使用first
函数获取每个分组中Column2的第一个元素,并使用last
函数获取每个分组中Column2的最后一个元素。最后,使用reset_index
函数重置索引,并将结果打印出来。