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

Question

我有一个熊猫数据test_df 。 此数据框包含 20 列。 例如，我得到了一个列索引col_index = [3, 5]的列表。

我需要创建两个单独的数据框

• 一个只包括 col_index 中的列
• other 包括除 col_index 中的列之外的所有列

我怎么做？

我明白我能做到

new_df = df.iloc[:, 3] 

从第 3 列中创建一个数据框。但是在这种情况下我该怎么做，因为我有多个列号要从主数据框中分离出来？

使用蟒蛇 3

Answer 1

你可以用drop

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