[Python] 20. Pandas 데이터프레임 값 일부를 추출해서 새로운 열 추가하기

Pandas 데이터프레임에서 일부 값을 추출하여 새로운 열을 추가하는 방법은 다양한 방법으로 가능합니다.
여기서는 몇 가지 일반적인 방법을 예시와 함께 설명하겠습니다.

예제 데이터프레임

import pandas as pd

data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Score': [90, 80, 70]
}

df = pd.DataFrame(data)
print(df)

단일 열 값 추출하여 새로운 열 추가하기

예를 들어, “Name” 열에서 첫 번째 글자만 추출하여 “Initial”이라는 새로운 열을 추가하고 싶다면 다음과 같이 할 수 있습니다.

df['Initial'] = df['Name'].str[0]
print(df)

출력

       Name  Score Initial
0     Alice     90       A
1       Bob     80       B
2   Charlie     70       C

여러 열 값 조합하여 새로운 열 추가하기

예를 들어, “Name”과 “Score” 열의 값을 조합하여 “Result”라는 새로운 열을 추가하고 싶다면 다음과 같이 할 수 있습니다.

df['Result'] = df['Name'] + ' - ' + df['Score'].astype(str)
print(df)

출력

       Name  Score           Result
0     Alice     90     Alice - 90
1       Bob     80       Bob - 80
2   Charlie     70   Charlie - 70

조건에 따라 새로운 열 추가하기

조건에 따라 새로운 열을 추가하고 싶다면 numpy.where()를 사용하여 조건을 지정할 수 있습니다.

import numpy as np

df['Pass/Fail'] = np.where(df['Score'] >= 80, 'Pass', 'Fail')
print(df)

출력

       Name  Score           Result Pass/Fail
0     Alice     90     Alice - 90      Pass
1       Bob     80       Bob - 80      Pass
2   Charlie     70   Charlie - 70      Fail

위 예제에서는 각각 단일 열 값을 추출하거나 여러 열 값을 조합하여 새로운 열을 추가하는 방법과, 조건에 따라 새로운 열을 추가하는 방법을 보여드렸습니다.
이러한 방법들을 활용하여 데이터프레임에 원하는 정보를 추가하거나 가공할 수 있습니다.

[Python] 21. Pandas 데이터프레임 잘 알려지지 않은 유용한 기능

파이썬의 pandas 라이브러리는 데이터프레임을 다루는데 매우 강력하고 유용한 기능을 많이 제공합니다. 다음은 pandas 데이터프레임에서 사람들이 모르거나 덜 주목하는 유용한 기능 몇 가지를 예제와 함께 상세하게 설명하겠습니다.…

더 보기 »

Related posts:

[Python] 21. Pandas 데이터프레임 잘 알려지지 않은 유용한 기능 [Python] 30. Pandas 데이터프레임 lambda 함수 사용하기 [Python] 18. pandas 데이터프레임 복사하는 방법 [Python] 24. Pandas 데이터프레임 IF 문을 사용하는 다양한 예제