删除 pandas 列 - Python

在 pandas 中，我们可以使用 "drop()" 函数来删除列，也可以使用 "del" 关键字。

使用 drop() 函数删除列

我们可以使用 "drop()" 函数来删除 pandas DataFrame 中的一列或多列。下面是一个例子：

import pandas as pd

# 创建一个 DataFrame
data = {'name': ['Tom', 'Jack', 'Steve', 'Ricky'], 
        'age': [28, 34, 29, 42], 
        'city': ['Beijing', 'New York', 'Sydney', 'London']}

df = pd.DataFrame(data)

# 删除一列
df = df.drop('city', axis=1)
print(df)

输出：

    name  age
0    Tom   28
1   Jack   34
2  Steve   29
3  Ricky   42

在上面的例子中，我们使用 "drop()" 函数删除了 DataFrame 中的 "city" 列，并指定 "axis=1" 表示删除列。

如果想删除多列，可以使用列表将这些列名传递给 "drop()" 函数，如下所示：

# 删除多列
df = df.drop(['age', 'name'], axis=1)
print(df)

输出：

Empty DataFrame
Columns: []
Index: [0, 1, 2, 3]

在上面的例子中，我们删除了整个 DataFrame，因为我们删除了所有列。

使用 del 关键字删除列

除了使用 "drop()" 函数之外，还可以使用 Python 的 "del" 关键字来删除列。下面是一个例子：

# 创建一个 DataFrame
data = {'name': ['Tom', 'Jack', 'Steve', 'Ricky'], 
        'age': [28, 34, 29, 42], 
        'city': ['Beijing', 'New York', 'Sydney', 'London']}

df = pd.DataFrame(data)

# 删除一列
del df['city']
print(df)

输出：

    name  age
0    Tom   28
1   Jack   34
2  Steve   29
3  Ricky   42

在上面的例子中，我们使用 "del" 关键字删除了 DataFrame 中的 "city" 列。

如果想删除多列，可以使用多个 "del" 语句，或者使用 "for" 循环来删除。

# 删除多列
del df['name']
del df['age']
print(df)

输出：

Empty DataFrame
Columns: []
Index: [0, 1, 2, 3]

在上面的例子中，我们通过多个 "del" 语句删除了所有列。