Pandas通过columns属性访问、修改和删除列

2023-02-24 00:00:00 修改 删除 属性

在 pandas 中,DataFrame 的列可以使用 columns 属性进行访问、修改或删除。

以下是使用 columns 属性访问 DataFrame 列的示例代码:

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 访问 DataFrame 列
print(df.columns)

输出结果如下:

Index(['A', 'B', 'C'], dtype='object')

在上面的示例代码中,columns 属性返回一个 Index 对象,其中包含 DataFrame 的所有列标签。可以像访问列表一样使用索引访问 Index 中的元素,例如 df.columns[0] 返回第一个列标签 'A'。

以下是使用 columns 属性修改 DataFrame 列的示例代码:

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 修改 DataFrame 列标签
df.columns = ['X', 'Y', 'Z']

# 打印 DataFrame 列
print(df.columns)

输出结果如下:

Index(['X', 'Y', 'Z'], dtype='object')

在上面的示例代码中,将 DataFrame 的列标签从 ['A', 'B', 'C'] 修改为 ['X', 'Y', 'Z']。

以下是使用 columns 属性删除 DataFrame 列的示例代码:

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 删除 DataFrame 列
df = df.drop(columns=['B'])

# 打印 DataFrame 列
print(df.columns)

输出结果如下:

Index(['A', 'C'], dtype='object')

在上面的示例代码中,使用 drop() 函数删除了 DataFrame 的 B 列,然后使用 columns 属性打印了删除后的列标签。

相关文章