从数据框中删除列 - Python

在处理数据时，有时候需要删除数据框中的某些列，这时需使用 Python 中的 pandas 库提供的函数。

方法

我们可以使用 drop 函数来删除数据框中的某些列：

df.drop(columns=["column1", "column2"], inplace=True)

解释一下以上代码中的参数：

• columns：指定需要删除的列名，以列表形式传入。
• inplace：指定是否修改原始数据，默认为 False，即不修改原始数据。如果需要修改原始数据，需将其设置为 True。

示例

我们准备一个示例数据，以便进行后续操作：

import pandas as pd

data = {
    "name": ["Tom", "Jack", "Mary", "Mike"],
    "age": [32, 28, 19, 25],
    "score1": [80, 90, 85, 75],
    "score2": [72, 88, 91, 79],
    "score3": [88, 82, 90, 76]
}

df = pd.DataFrame(data)

这里我们定义了一个包含五列数据的数据框。

1. 删除单列

假设我们想要删除 "score1" 列，可以使用以下代码：

df.drop(columns=["score1"], inplace=True)

上面代码中，我们指定了需要删除的列名为 "score1"，同时将 inplace 参数设置为 True，即修改原始数据。

2. 删除多列

现在我们想要同时删除 "score2" 和 "score3" 两列，可以使用以下代码：

df.drop(columns=["score2", "score3"], inplace=True)

以上代码中，我们指定了需要删除的列名为 "score2" 和 "score3"，同时将 inplace 参数设置为 True，即修改原始数据。

结论

如上所示，我们可以使用 drop 函数来删除数据框中的某些列。此函数非常方便且易于使用，只需指定需要删除的列名即可。