[Python] 판다스, 넘파이 axis 개념

[Python] 판다스, 넘파이 axis 개념

1. axis=0

• 각 열(row)의 모든 행(column)에 대해 동작
• 진행방향이 세로임 (위에서 아래)
• drop일 때는 행을 drop

2. axis=1

• 각 행(column)의 모든 열(row)에 대해 동작
• 진행방향이 가로임 (왼쪽에서 오른쪽)
• drop일 때는 열을 drop

 

---

 

▶ 하나의 데이터 불러오기

import pandas as pd
import numpy as np

df = pd.read_csv("https://bit.ly/wine_csv_data").head()
df

 

▶ pd.drop에서 axis 예시

• 인덱스가 0, 3인 행을 삭제하기 (axis=0)

df.drop([0,3], axis=0)

 

• 'sugar' 열 삭제하기 (axis=1)

df.drop('sugar', axis=1)

 

 

▶ pd.mean()에서 axis 예시

• 'alcohol', 'sugar' 컬럼의 평균 구하기 (axis=0 이므로 세로 방향으로)

df[['alcohol', 'sugar']].mean(axis=0)

 

• 'alcohol', 'sugar' 컬럼의 평균 구하기 (axis=1 이므로 가로 방향으로)

df[['alcohol', 'sugar']].mean(axis=1)

 

▶ np.mean()에서 axis 예시

• 세로 방향으로 df의 평균 구하기 (axis=0)

np.mean(df, axis=0)

 

• 가로 방향으로 df의 평균 구하기 (axis=1)

np.mean(df, axis=1)

 

 

 

▶ 헷갈리지 말기!

1. drop 할 때는 axis=0이면 행을 삭제, axis=1이면 열을 삭제
2. pd.mean, np.mean에서는 axis=0이면 세로로 계산(행 방향), axis=1이면 가로로 계산(열 방향)
   • axis=0일 때 행 별로 평균 계산하라는 의미가 아님!
   • axis=1일 때 열 별로 평균 계산하라는 의미가 아님!