根据输入列索引拆分和合并熊猫数据框

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()