Использование переменной для ссылки на имя колонки Panda в python
я пытаюсь написать функцию, чтобы упростить код, в результате я передаю переменные, содержащие имена столбцов. это для приложения Django, и отладчик не дает никакой информации о том, где моя ошибка, только "внутренняя ошибка сервера".
Мой код работает нормально, написанный НЕ как функция:
df_trips['trip_time_prep_starts'] = df_trips["trip_time_prep_starts"].map(str)
df_trips.trip_time_prep_starts = pd.to_datetime(df_trips.trip_time_prep_starts)
df_trips['trip_time_left_house'] = df_trips['trip_time_left_house'].map(str)
df_trips.trip_time_left_house = pd.to_datetime(df_trips.trip_time_left_house)
df_trips['stage1_duration']=(df_trips['trip_time_left_house']-df_trips['trip_time_prep_starts']).dt.total_seconds()
записанная как функция, передающая следующие переменные:
stage_name="stage1_duration"
stage_start="trip_time_prep_starts"
stage_end="trip_time_left_house"
моя функция:
def calc_stage_duration(df_trips, stage_name, stage_start, stage_end):
df_trips[stage_start] = df_trips[stage_start].map(str)
df_trips.[stage_start] = pd.to_datetime(df_trips.[stage_start])
df_trips[stage_end] = df_trips[stage_end].map(str)
df_trips.[stage_end] = pd.to_datetime(df_trips.[stage_end])
df_trips[stage_name]=(df_trips[stage_end]-df_trips[stage_start]).dt.total_seconds()
Я не могу найти свою ошибку и не понимаю, как правильно использовать скобки и точки, например, df_trip.[stage_end] против df_trips[stage_end]