[python] axis=0 axis=1

매번 헷갈려서 적어보는 포스팅..........................

 

axis=1은 행방향(가로방향) 인데 drop함수(열방향)에서만 예외다 라고 외워야 겠다................

 

 

import pandas as pd
df = pd.DataFrame([[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3]], columns=["col1", "col2", "col3", "col4"])
df

 

# mean, sum, max, min 등등 에서 axis=1 은 행방향(가로방향)

df.mean(axis=1) # 행의 평균 (가로)

0    1.0
1    2.0
2    3.0
dtype: float64

 

# drop 에서 axis=1 은 열방향(세로방향)

df.drop('col1', axis=1) # 열의 삭제 (세로)

   col2  col3  col4
0     1     1     1
1     2     2     2
2     3     3     3