판다스로 여러 열 추가하는 방법 - 초보자를 위한 간편한 가이드

[

판다스(Pandas)로 여러 개의 열(Columns) 추가하기

소개

판다스(Pandas)는 파이썬에서 데이터 조작과 분석에 사용되는 강력한 라이브러리입니다. 이 튜토리얼에서는 판다스를 사용하여 데이터프레임(DataFrame)에 여러 개의 열을 추가하는 방법에 대해 자세히 알아보겠습니다.

요약

판다스를 사용하여 데이터프레임에 여러 개의 열을 추가하는 방법을 배웁니다. 첫 번째로 열을 추가하는 일반적인 방법과 두 번째로 데이터프레임에 조건에 따라 여러 개의 열을 추가하는 방법을 알아봅니다. 두 가지 방법은 모두 판다스의 기능을 사용하여 실행됩니다.

1. 열 추가 - 일반적인 방법

일반적인 방법으로 데이터프레임에 열을 추가하는 방법입니다. DataFrame의 insert 메서드를 사용하여 열을 추가할 수 있습니다. 예를 들어, 새로운 열 “age”를 “data” 데이터프레임의 2번째 위치에 추가하려면 다음과 같이 작성할 수 있습니다.

import pandas as pd

data = {'name': ['John', 'Steve', 'Sarah'],
        'gender': ['Male', 'Male', 'Female']}
df = pd.DataFrame(data)

df.insert(1, 'age', [25, 32, 28])

새로운 열을 추가하기 위해 insert 함수의 첫 번째 인자로 열의 위치를 지정하고, 두 번째 인자로 열의 이름을 설정하고, 세 번째 인자로 열에 할당할 값을 입력합니다. 이 예제에서는 “age” 열을 1번 위치에 추가하였습니다.

2. 열 추가 - 조건에 따라 여러 개의 열 추가하기

조건에 따라 여러 개의 열을 추가하기 위해서는 조건에 맞는 값을 계산한 후, 그 값을 열로 추가해야 합니다. 이를 위해 판다스의 apply 메서드를 사용합니다.

먼저, 기존 데이터프레임에 추가할 열의 값을 계산하는 함수를 작성합니다. 그리고 apply 메서드를 사용하여 이 함수를 호출하고 계산 결과를 데이터프레임에 새로운 열로 추가합니다. 예제를 확인해보겠습니다.

import pandas as pd

data = {'name': ['John', 'Steve', 'Sarah'],
        'gender': ['Male', 'Male', 'Female'],
        'age': [25, 32, 28]}
df = pd.DataFrame(data)

def calculate_score(row):
    if row['gender'] == 'Male':
        return row['age'] * 2
    else:
        return row['age'] * 3

df['score'] = df.apply(calculate_score, axis=1)

위의 예제에서는 “gender” 열 값이 ‘Male’인 경우 “age” * 2로, 그렇지 않은 경우 “age” * 3으로 “score” 열 값을 계산한 후 데이터프레임에 추가하였습니다.

FAQ (자주 묻는 질문)

Q1: 어떻게 판다스 데이터프레임에 여러 개의 열을 한 번에 추가할 수 있을까요?

A1: 판다스로 여러 개의 열을 한 번에 추가하는 일반적인 방법은 ‘insert’ 함수를 사용하는 것입니다. ‘insert’ 함수의 첫 번째 인자로 열의 위치, 두 번째 인자로 열의 이름, 세 번째 인자로 열에 할당할 값을 입력합니다.

Q2: 조건에 따라 여러 개의 열을 추가하는 방법은 무엇인가요?

A2: 조건에 따라 여러 개의 열을 추가하기 위해서는 apply 메서드를 사용해야 합니다. 함수를 작성하여 조건에 맞는 값을 계산한 후 apply 메서드를 사용하여 이 값을 열로 추가할 수 있습니다.

Q3: 왜 `apply` 메서드의 `axis`를 1로 설정해야 하나요?

A3: apply 메서드를 사용하여 함수를 적용할 축을 지정해야 합니다. 1을 설정하면 함수가 데이터프레임의 각 행에 대해 적용되고, 0을 설정하면 함수가 데이터프레임의 각 열에 대해 적용됩니다.

Q4: 열을 추가할 때 기존 열의 값을 참조할 수 있나요?

A4: 네, 열을 추가할 때 기존 열의 값을 참조할 수 있습니다. 함수 안에서 기존 열의 값을 참조하여 새로운 값을 계산할 수 있습니다.

Q5: 기존 열을 수정하지 않고 새로운 열만 추가할 수 있을까요?

A5: 기존 열을 수정하지 않고 새로운 열만 추가하려면 ‘assign’ 함수를 사용할 수 있습니다. ‘assign’ 함수는 기존 데이터프레임을 수정하지 않고 새로운 데이터프레임을 반환합니다.