разделить и объединить кадр данных pandas на основе индекса входного столбца

Question

У меня есть датафрейм для панд test_df. Этот фрейм данных содержит 20 столбцов. Мне дан список столбцов индекса col_index = [3, 5], например.

Мне нужно создать два отдельных кадра данных

• только один, включая столбцы в col_index
• other включает все столбцы, кроме столбцов в col_index

Как мне это сделать?

Я понимаю, что могу сделать

new_df = df.iloc[:, 3] 

Чтобы создать фрейм данных из столбца № 3. Но что мне делать, так как в этом случае у меня есть несколько номеров столбцов, чтобы отделить их от основного фрейма данных?

Использование Python 3

Answer 1

Вы можете сделать с drop

df1=df.iloc[:,col_index].copy()
df2=df.drop(df1.columns.tolist(),axis=1).copy()