데이터프레임 axis 방향이 헷갈리네요

안녕하세요. 마지막 잎 경진대회에서 사용했던 apply 함수인데

여기서, train_df의 shape이 (행, 열)인데, 저는 개인적으로 axis=0이면 행 방향마다 데이터를 추출할 줄 알았는데, 그 반대더군요. 어떻게 구분하게 되나요? 헷갈리네요

def preprocessing(x):
    if x['healthy'] == 1:
        return 'healthy'
    elif x['multiple_diseases'] == 1:
        return 'multiple_diseases'
    elif x['rust'] == 1:
        return 'rust'
    elif x['scab'] == 1:
        return 'scab'
    else:
        return 'None'

train_df['label'] = train_df.apply(lambda x : preprocessing(x), axis=1)
train_df