Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Pandas DataFrame에서 열의 첫 글자를 대문자로 변환하는 방법

이 튜토리얼에서는 Pandas DataFrame의 특정 열에서 첫 글자만 대문자로 변환하는 방법을 알아보겠습니다. 여기서는 문자열(str)의 capitalize() 메서드를 활용합니다.

먼저 파이썬 리스트로부터 DataFrame을 생성한 뒤, str.capitalize() 메서드를 적용해 작업을 완료하는 과정을 살펴보겠습니다.

예제 1: DataFrame 생성하기

# pandas 라이브러리 임포트
import pandas as pd

# DataFrame 열에 사용할 리스트
names = ['tutorialspoint', 'mohit', 'sharma']
age = [25, 34, 21]

# DataFrame 생성
data_frame = pd.DataFrame({'Name': names, 'Age': age})

# DataFrame 확인
print(data_frame)

출력 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

            Name   Age
0  tutorialspoint    25
1           mohit    34
2          sharma    21

예제 2: 첫 글자 대문자로 변환하기

이제 Name 열의 각 값에서 첫 글자를 대문자로 변환해 보겠습니다. 작업 순서는 다음과 같습니다.

  • 열 이름(인덱스)으로 해당 열을 선택한 후 capitalize() 메서드를 적용합니다.
  • 변환된 결과를 동일한 열에 다시 할당합니다.

실제 코드로 확인해 보겠습니다.

# pandas 라이브러리 임포트
import pandas as pd

# DataFrame 열에 사용할 리스트
names = ['tutorialspoint', 'mohit', 'sharma']
age = [25, 34, 21]

# DataFrame 생성
data_frame = pd.DataFrame({'Name': names, 'Age': age})

# Name 열의 첫 글자를 대문자로 변환 후 다시 할당
data_frame['Name'] = data_frame['Name'].str.capitalize()

# DataFrame 출력
print(data_frame)

출력 결과

위 코드를 실행하면 다음과 같이 Name 열의 첫 글자가 모두 대문자로 변경된 것을 확인할 수 있습니다.

            Name   Age
0  Tutorialspoint    25
1           Mohit    34
2          Sharma    21

참고 사항

물론 사용자 정의 함수나 apply(), map() 등의 메서드를 조합해서도 같은 결과를 얻을 수 있습니다. 하지만 str.capitalize() 메서드를 사용하는 것이 가장 간결하고 직관적이며, 코드 가독성 면에서도 유리합니다.

또한 참고로, 각 단어의 첫 글자를 모두 대문자로 만들고 싶다면 str.title(), 전체 문자열을 대문자로 변환하려면 str.upper() 메서드를 사용하면 됩니다.

결론

이번 튜토리얼에서는 Pandas DataFrame의 열에서 첫 글자를 대문자로 변환하는 방법을 배웠습니다. 튜토리얼에 대해 궁금한 점이 있다면 댓글로 남겨주세요.