IT/python

[파이썬] dataframe 새로운 컬럼 추가

generator 2024. 11. 19. 18:14

디비에서 조회한 데이터를 가공할 일이 생겨서 아래처럼 처리를 했습니다.

dt.at[index,'컬럼명'] 에대 넣어 주야만 정상적으로 데이터가 들어갑니다.

# 빈 컬럼 추가 (갱신할 값)
df['new_value'] = None

# 각 code 값으로 추가 데이터 조회
for index, row in df.iterrows():
    code = row['code']
    
    # 두 번째 조회
    query2 = f"SELECT other_value FROM another_table WHERE code = '{code}'"
    result = pd.read_sql(query2, connection)
    
    # 결과가 있으면 갱신
    if not result.empty:
        df.at[index, 'new_value'] = result.loc[0, 'other_value']

print("갱신된 데이터")
print(df)