可以使用 DataFrame 的 iloc 属性来提取部分列。
df.iloc[:,[2, 3, 4, 5, TARGET_COLUMN]]
或者
df[df.columns[[2, 3, 4, 5, TARGET_COLUMN]]]
如果要根据列名提取列,可以使用 DataFrame 的 loc 属性。
df[['column_name1','column_name2']]
注意: 在上面的代码中,需要替换 TARGET_COLUMN 为目标列的名称或索引值。
可以使用 DataFrame 的 iloc 属性来提取部分列。
df.iloc[:,[2, 3, 4, 5, TARGET_COLUMN]]
或者
df[df.columns[[2, 3, 4, 5, TARGET_COLUMN]]]
如果要根据列名提取列,可以使用 DataFrame 的 loc 属性。
df[['column_name1','column_name2']]
注意: 在上面的代码中,需要替换 TARGET_COLUMN 为目标列的名称或索引值。