Pandas Dataframe: 保留特定列 - 只保留特征1到特征9

Pandas Dataframe: 保留特定列 - 只保留特征1到特征9

在使用 Pandas 处理数据时，我们经常需要从 DataFrame 中选择特定的列。例如，我们想要保留特征1到特征9作为数据集内容，而删除其他列。

错误代码:

X = data.drop(labels=['特征1', '特征2', '特征3', '特征4', '特征5', '特征6', '特征7', '特征8','特征9'], axis=1)

这段代码使用 drop 方法删除了特征1到特征9，这与我们的目标相反。

正确代码:

X = data[['特征1', '特征2', '特征3', '特征4', '特征5', '特征6', '特征7', '特征8','特征9']]

通过在方括号内列出我们想要保留的列名，我们可以轻松地从 DataFrame 中选择特定列。

示例:

假设我们有一个名为 data 的 DataFrame，它包含以下列：

['特征1', '特征2', '特征3', '特征4', '特征5', '特征6', '特征7', '特征8', '特征9', '特征10', '特征11']

使用 X = data[['特征1', '特征2', '特征3', '特征4', '特征5', '特征6', '特征7', '特征8','特征9']] 我们可以得到一个新的 DataFrame X，它只包含特征1到特征9。

总结:

要从 Pandas DataFrame 中选择特定列，可以使用方括号内列出想要保留的列名。这比使用 drop 方法更加直观和高效。