프로그래밍/파이썬(Python)

[Pandas] 대괄호([])를 2번 사용하는 이유(단일 열/2개 이상의 열)

정정훈훈 2024. 9. 22. 14:51
반응형

print(sample[['나', '라']])에서 대괄호를 두 번 사용하는 이유는 다음과 같습니다.

  1. sample['나']:
    • sample['나']는 단일 열을 선택하는 방식입니다. 이렇게 하면 pandas.Series 객체가 반환됩니다. 즉, 하나의 열만 가져오는 경우입니다.
  2. sample[['나', '라']]:
    • sample[['나', '라']]는 여러 열을 선택하는 방식입니다. 대괄호 안에 리스트 ['나', '라']를 넣어, 두 개 이상의 열을 선택하는 것입니다.
    • 이렇게 하면 pandas.DataFrame 객체가 반환됩니다. 즉, 여러 열을 가져오므로 DataFrame 형태를 유지합니다.

따라서, 대괄호가 한 번 사용되면 단일 열이 선택되어 Series가 반환되고, 두 번 사용되면 열 목록이 선택되어 DataFrame이 반환됩니다.

정리:

  • sample['나']: 단일 열 선택 → Series 반환
  • sample[['나', '라']]: 여러 열 선택 → DataFrame 반환
반응형