pandas lamda 예제

pandas에서 lambda 함수를 사용하는 예제를 알려드리겠습니다.

1. 데이터프레임의 열에 lambda 함수 적용하기

import pandas as pd

# 샘플 데이터프레임 생성
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})

# 열 'A'에 lambda 함수 적용하기
df['C'] = df['A'].apply(lambda x: x * 2)

print(df)

출력:

   A  B  C
0  1  4  2
1  2  5  4
2  3  6  6

2. 데이터프레임 정렬하기:

import pandas as pd

# 샘플 데이터프레임 생성
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})

# lambda 함수를 사용하여 데이터프레임 정렬하기
df = df.sort_values(by=['A', 'B'], key=lambda x: [x.str.lower()])

print(df)

출력:

   A  B
0  1  4
1  2  5
2  3  6

3. 데이터프레임 필터링하기:

import pandas as pd

# 샘플 데이터프레임 생성
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})

# lambda 함수를 사용하여 데이터프레임 필터링하기
filtered_df = df[df['A'].apply(lambda x: x > 1)]

print(filtered_df)

출력:

   A  B
1  2  5
2  3  6

이처럼 lambda 함수는 pandas에서 다양한 방식으로 활용될 수 있습니다. 간단한 함수를 빠르게 정의할 수 있어 코드를 더 간결하게 만들 수 있습니다.