从数据框中删除列 - Python

在Python中的pandas库中，可以方便地创建和操作数据框，但有时候需要在数据框中删除某些列。

以下是如何在Python中用pandas删除数据框中的列的示例。

1. 准备数据

我们将使用pandas的read_csv函数从csv文件中读取数据作为演示。

import pandas as pd

# 从csv文件中读取数据
df = pd.read_csv('data.csv')

# 显示数据框前5行
print(df.head())

这将输出数据框的前5行数据。

2. 删除列

方法1 - 使用drop函数

可以使用drop函数从数据框中删除列。drop函数需要传入列名称作为参数。

# 从数据框中删除列
df = df.drop('列名称', axis=1)

# 显示数据框前5行
print(df.head())

这可以从数据框中删除名为'列名称'列。

方法2 - 通过列索引删除列

如果您知道要删除的列的索引，可以通过以下方式从数据框中删除列。

# 通过索引从数据框中删除列
df = df.drop(df.columns[[0, 1]], axis=1)

# 显示数据框前5行
print(df.head())

这将从数据框中删除以0和1为索引的两列。

3. 保存更改

如果您想要保存更改的数据框，可以使用to_csv函数将其写入csv文件中。

# 将更改后的数据框写入csv文件
df.to_csv('new_data.csv', index=False)

这将把更改后的数据框写入'new_data.csv'文件中，并忽略索引。

扩展阅读

如果您想了解更多有关pandas的信息，请参阅pandas官方文档。

结论

在Python中使用pandas库可以轻松创建和操作数据框，并使用drop函数从数据框中删除列。同时，使用to_csv函数将更改后的数据框写入csv文件也很简单。