在 DataFrame 中获取列名和行名

在数据分析中，我们通常会使用 Pandas 库中的 DataFrame 数据结构进行数据处理和分析。而在 DataFrame 中获取列名和行名是一个比较基础的操作，本文将向大家介绍如何在 Pandas DataFrame 中获取列名和行名。

获取列名

我们可以使用 DataFrame 的 columns 属性获取所有的列名，该属性返回的是一个 Index 对象，可以使用 tolist() 方法将其转换为列表类型。

import pandas as pd

# 创建 DataFrame
data = {'name': ['Tom', 'Jerry', 'Mickey'],
        'age': [25, 30, 28],
        'city': ['Shanghai', 'New York', 'Paris']}
df = pd.DataFrame(data)

# 获取列名
columns = df.columns.tolist()
print(columns)

上述代码将输出以下结果：

['name', 'age', 'city']

获取行名

我们可以使用 DataFrame 的 index 属性获取所有的行名，该属性返回的也是一个 Index 对象，同样可以使用 tolist() 方法将其转换为列表类型。

import pandas as pd

# 创建 DataFrame
data = {'name': ['Tom', 'Jerry', 'Mickey'],
        'age': [25, 30, 28],
        'city': ['Shanghai', 'New York', 'Paris']}
df = pd.DataFrame(data)

# 获取行名
index = df.index.tolist()
print(index)

上述代码将输出以下结果：

[0, 1, 2]

获取特定行或列的名称

如果我们要获取特定行或列的名称，可以通过选取行或列的方式获取。比如，要获取第一列的名称，可以使用 df.columns[0] 或 df.columns.tolist()[0]。

import pandas as pd

# 创建 DataFrame
data = {'name': ['Tom', 'Jerry', 'Mickey'],
        'age': [25, 30, 28],
        'city': ['Shanghai', 'New York', 'Paris']}
df = pd.DataFrame(data)

# 获取第一列的名称
print(df.columns[0])  # 输出 'name'
print(df.columns.tolist()[0])  # 输出 'name'

# 获取第一行的名称
print(df.index[0])  # 输出 0
print(df.index.tolist()[0])  # 输出 0

上述代码将输出以下结果：

name
name
0
0

通过本文的介绍，我们学习了如何在 Pandas DataFrame 中获取列名和行名，掌握了选取特定行或列名称的方法。这些操作是我们后续数据分析和处理的基础。