python apply 함수 응용

1)
Series 객체에는 name 이라는 field가 있음
df.apply(lambda x:np.square(x) if x.name in ['A','B'] else x)
# 이렇게 하면 컬럼의 이름이 'A' 또는 'B' 인 경우에만 함수를 적용한다

2) 홀수 행에만 적용
df.apply(lambda x:np.square(x) if (x.name+1)%2==1, axis=1)

3) 함수에 인자 받기
def cal1(x,a,b):
  return a*x-b
df.apply(cal1, args(2,3))