pandas 특정열 다음에 열추가

import pandas as pd

# 초기 데이터프레임
data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
}
df = pd.DataFrame(data)

# 새로운 열 D를 추가할 위치 계산 (열 B 다음)
col_position = df.columns.get_loc('B') + 1

# B 열이 마지막 열이라면 맨 마지막에 추가
if col_position >= len(df.columns):
    df['D'] = 10
else:
    df.insert(col_position, 'D', 10)

print(df)

import pandas as pd

# 초기 데이터프레임
data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
}
df = pd.DataFrame(data)

# 새로운 열 D를 추가할 위치 계산 (열 B 다음)
col_position = df.columns.get_loc('B') + 1

# 새로운 열 D를 삽입
df.insert(col_position, 'D', 10)

print(df)