博客

如何在DataFrame中获得列和行的名称

  • Post category:Python

要在 Pandas 的 DataFrame 中获取列和行的名称，可以使用以下方法：

  • 获取列名：使用 df.columns 属性
  • 获取行名：使用 df.index 属性

下面我们来分别详细讲解。

获取列名

对于 DataFrame 中的列，我们可以使用 df.columns 属性来获取它们的名称。下面是一个实例：

import pandas as pd

# 创建一个表格
data = {
    'name': ['Alice', 'Bob', 'Charlie', 'David'],
    'age': [25, 32, 18, 47],
    'city': ['Beijing', 'Shanghai', 'New York', 'Tokyo']
}
df = pd.DataFrame(data)

# 获取所有列名
columns = df.columns

# 打印出列名
print(columns)

执行上述代码后，输出的结果如下：

Index(['name', 'age', 'city'], dtype='object')

其中，Index 表示得到的是一个 Index 对象，里面包含了所有的列名。我们可以用 list() 函数将它们转化为列表。

# 转化为列表
columns = list(df.columns)
print(columns)

输出结果仍旧为：

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

获取行名

与获取列名类似，我们可以使用 df.index 属性来获取 DataFrame 的行名。下面是一个实例：

import pandas as pd

# 创建一个表格
data = {
    'name': ['Alice', 'Bob', 'Charlie', 'David'],
    'age': [25, 32, 18, 47],
    'city': ['Beijing', 'Shanghai', 'New York', 'Tokyo']
}
df = pd.DataFrame(data)

# 获取所有行名
index = df.index

# 打印出行名
print(index)

执行上述代码后，输出的结果如下：

RangeIndex(start=0, stop=4, step=1)

其中，RangeIndex(start=0, stop=4, step=1) 表示行名为 0,1,2,3。同样，我们可以用 list() 函数将它们转化为列表。

# 转化为列表
index = list(df.index)
print(index)

输出结果为：

[0, 1, 2, 3]

以上就是在 Pandas 的 DataFrame 中获取列和行的名称的详细攻略，希望对你有所帮助。

Tags: