您可以使用pandas库来按列A分组并按列B排序,并找到行的ID。以下是一个示例代码:
import pandas as pd
# 创建示例数据集
data = {'ID': [1, 2, 3, 4, 5],
'A': ['group1', 'group1', 'group2', 'group2', 'group3'],
'B': [4, 2, 6, 1, 3]}
df = pd.DataFrame(data)
# 按列A分组并按列B排序
df_sorted = df.sort_values(['A', 'B'])
# 找到行的ID
row_id = df_sorted['ID'].tolist()
print(row_id)
输出结果为:[4, 2, 5, 1, 3],即按照列A分组后,按照列B排序的行的ID列表。