파이썬 Pandas DataFrame .filter() items 옵션

판다스(Pandas)의 DataFrame에서 filter() 함수는 열(axis=1) 또는 행(axis=0)을 선택하는 데 사용됩니다. filter() 함수는 기본적으로 열(axis=1)을 선택하는 함수입니다.

먼저, DataFrame.filter() 함수는 items 옵션을 사용하여 열의 이름을 지정하여 열을 선택할 수 있습니다. 이 경우, 열 이름을 문자열 또는 리스트 형식으로 전달할 수 있습니다.

예를 들어, 다음과 같은 DataFrame이 있다고 가정해 봅시다.

import pandas as pd

df = pd.DataFrame({'이름': ['Alice', 'Bob', 'Charlie', 'Dave'],
                   '나이': [25, 30, 35, 40],
                   '성별': ['여성', '남성', '남성', '남성'],
                   '연봉': [5000, 6000, 7000, 8000]})

728x90

이제 filter() 함수를 사용하여 '이름' 열을 선택해 보겠습니다.

# '이름' 열을 선택
name_col = df.filter(items=['이름'])
print(name_col)

# Output
#         이름
# 0     Alice
# 1       Bob
# 2   Charlie
# 3      Dave

위의 코드에서는 DataFrame의 filter() 함수에 items 옵션을 사용하여 '이름' 열을 선택했습니다. 이를 출력하면 '이름' 열만 있는 DataFrame이 반환됩니다.

이와 같이, filter() 함수의 items 옵션을 사용하여 열을 선택할 수 있습니다. 이를 활용하면 데이터 분석에서 원하는 열을 쉽게 선택할 수 있습니다.