[Pandas tips] 행의 평균값 정렬

소스 코드

원래 mean열을 만들었는데 df.mean(axis=1).sort_values(ascending=False).keys() 순서를 직접 추출할 수 있습니다.좀 길어서 함수 형태로 공개합니다.

def sort_row_mean(df):
    return df.loc[df.mean(axis=1).sort_values(ascending=False).keys(), :]

예제

아이리스 데이터 집합의 평균치가 높은 순서에 따라 정렬해 보세요.

소스 코드

import pandas as pd
from sklearn import datasets

def sort_row_mean(df):
    return df.loc[df.mean(axis=1).sort_values(ascending=False).keys(), :]


iris = datasets.load_iris()
df = pd.DataFrame(iris.data, columns=iris.feature_names)

sort_row_mean(df)

출력

sepal length (cm)
sepal width (cm)
petal length (cm)
petal width (cm)
117
7.7
3.8
6.7
2.2
131
7.9
3.8
6.4
2.0
118
7.7
2.6
6.9
2.3
109
7.2
3.6
6.1
2.5
105
7.6
3.0
6.6
2.1
...
...
...
...
...
42
4.4
3.2
1.3
0.2
8
4.4
2.9
1.4
0.2
38
4.4
3.0
1.3
0.2
13
4.3
3.0
1.1
0.1
41
4.5
2.3
1.3
0.3