랭귀지/SWIFT

pandas 특정열 값채우기

유키공 2024. 6. 13. 21:17
import pandas as pd

# 예시 데이터프레임 생성
data = {
    'A': [1, 2, 3, 4, 5],
    'B': [6, 7, 8, 9, 10],
    'C': [11, 12, 13, 14, 15],
    'D': [16, 17, 18, 19, 20],
    'E': [21, 22, 23, 24, 25]
}
df = pd.DataFrame(data)

# 3개의 열 선택 (예: 'B', 'C', 'E')
cols_to_zero = ['B', 'C', 'E']

# 선택한 열을 0으로 채우기
df[cols_to_zero] = 0

print(df)
import pandas as pd
import numpy as np

# 예시 데이터프레임 생성 (NaN 포함)
data = {
    'A': [1, 2, np.nan, 4, 5],
    'B': [6, np.nan, 8, 9, 10],
    'C': [11, 12, 13, np.nan, 15],
    'D': [16, 17, 18, 19, 20],
    'E': [21, 22, 23, 24, np.nan]
}
df = pd.DataFrame(data)

# 3개의 열 선택 (예: 'B', 'C', 'E')
cols_to_fill = ['B', 'C', 'E']

# 선택한 열의 NaN 값을 0으로 채우기
df[cols_to_fill] = df[cols_to_fill].fillna(0)

print(df)