파이썬 기초문법

[Python] DataFrame 컬럼 내 리스트 형식의 데이터의 모든 행에 인자를 추가하는 방법

Family in August 2023. 2. 24. 08:26
반응형

pandas DataFrame의 리스트 컬럼에 모든 행에 대해 인자를 append하는 방법은 다음과 같습니다.


import pandas as pd

# 샘플 데이터프레임 생성
df = pd.DataFrame({'name': ['John', 'Kate', 'Mike'], 'scores': [[10, 20], [30, 40], [50, 60]]})
print(df)

# 리스트 컬럼에 모든 행에 대해 인자 추가하기
new_score = 70
df['scores'] = df['scores'].apply(lambda x: x.tolist() + [new_score])
print(df)

위 코드에서는 먼저 샘플 데이터프레임을 생성합니다.
이 데이터프레임은 'name'과 'scores' 두 개의 컬럼으로 구성되어 있습니다. 'scores' 컬럼은 리스트 형식으로 각 행마다 두 개의 원소를 가지고 있습니다.

다음으로, 'scores' 컬럼에 모든 행에 대해 인자를 추가하는 부분입니다.
apply() 함수를 사용하여 모든 행에 대해 lambda 함수를 적용합니다.
이 lambda 함수에서는 각 행의 'scores' 리스트에 새로운 점수를 추가하고 반환합니다.
이렇게 반환된 값을 'scores' 컬럼에 할당하여 데이터프레임을 업데이트합니다.

위 코드를 실행하면, 'scores' 컬럼의 모든 행에 새로운 점수가 추가된 것을 확인할 수 있습니다.

반응형