Pandas DataFrame의 각 그룹에서 상위 레코드를 효율적으로 가져오려면 어떻게 해야 합니까?

Pandas: 그룹 내에서 효율적으로 최상위 레코드 얻기

Pandas DataFrames로 작업할 때 각 그룹에서 주요 레코드를 추출해야 하는 경우가 많습니다. . 일반적인 접근 방식은 'groupby' 및 'apply' 기능을 활용하여 각 그룹 내의 레코드를 열거하는 것입니다.

dfN = df.groupby('id').apply(lambda x:x['value'].reset_index()).reset_index()

그러나 보다 간소화된 접근 방식이 있습니다:

df.groupby('id').head(2)

이것은 메서드는 중간 계산 없이 최상위 레코드를 직접 가져옵니다. 또한 생성된 DataFrame은 원래 인덱스를 유지합니다.

결과 MultiIndex를 평면화하려면 다음을 사용하세요.

df.groupby('id').head(2).reset_index(drop=True)

이렇게 하면 다음 DataFrame이 생성됩니다.

id	value
1	1
1	2
2	1
2	2
3	1
4	1

또는 , SQL의 "row_number()" 창 함수를 사용하여 그룹 내의 레코드를 효율적으로 열거할 수 있습니다. 하지만 이 기능은 현재 Pandas에서 사용할 수 없습니다.

위 내용은 Pandas DataFrame의 각 그룹에서 상위 레코드를 효율적으로 가져오려면 어떻게 해야 합니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!