아미(아름다운미소)

pandas 빈값이 아닌면 a컬럼을 _기준으로 split 하고 첫번째값 사용 본문

랭귀지/pandas

pandas 빈값이 아닌면 a컬럼을 _기준으로 split 하고 첫번째값 사용

유키공 2025. 1. 3. 13:14
import pandas as pd

# 예시 데이터프레임 생성
data = {
    'a': ['value1_value2', '', 'value3_value4', 'value5_value6'],
    'b': [1, 2, 3, 4]
}
df = pd.DataFrame(data)

# a 컬럼이 빈 문자열이 아닐 경우 _ 기준으로 split하고 첫 번째 값 사용
df['first_value'] = df['a'].apply(lambda x: x.split('_')[0] if x else None)

print(df)
Comments