Python DataFrame 添加一列序号

在数据分析和处理过程中，常常需要给数据表（DataFrame）添加一些额外的信息，方便后续的操作和分析。在使用 pandas 库时，往往需要为一个 DataFrame 添加序号，以便于标识每一行数据。本文将介绍如何使用 Python 的 pandas 库，给 DataFrame 添加一列序号，以及相关的代码示例。

什么是 DataFrame？

DataFrame 是 pandas 库中用于存储表格数据的核心数据结构。它可以被视作一个二维数组，类似于 Excel 表格，由行和列组成。每一列可以包含不同的数据类型（例如整数、浮点数和字符串），而每一行代表一条记录。在数据分析中，DataFrame 是一种非常方便的数据表示方式。

如何添加序号列？

以下是实现将序号添加到 DataFrame 的步骤：

  1. 导入 pandas 库：首先确保您已经安装了 pandas 库，可以使用 pip 命令进行安装。
  2. 创建 DataFrame：构建一个简单的 DataFrame，您可以使用字典、列表等结构。
  3. 添加序号列：使用 DataFrame.insert() 方法或者直接通过赋值来新增一列序号。

下面是一个简单的代码示例，展示如何进行上述操作。

import pandas as pd

# 创建一个简单的 DataFrame
data = {
    '姓名': ['张三', '李四', '王五'],
    '年龄': [25, 30, 28],
    '性别': ['男', '女', '男']
}
df = pd.DataFrame(data)

# 显示原始 DataFrame
print("原始 DataFrame：")
print(df)

# 使用 insert() 方法添加序号列
df.insert(0, '序号', range(1, len(df) + 1))

# 显示添加序号后的 DataFrame
print("\n添加序号后的 DataFrame：")
print(df)

在上面的代码中，我们首先导入了 pandas 库，然后创建了一个包含姓名、年龄和性别的简单 DataFrame。接着，我们使用 insert() 方法将序号列添加到 DataFrame 的最前面。insert() 方法的第一个参数指定了要插入的位置（0 表示第一列），第二个参数是列名（'序号'），第三个参数是序号系列，使用 range() 函数生成从 1 到行数的序号。

结果分析

运行以上代码后，您会发现在原始 DataFrame 的最前面成功添加了一列序号。输出结果如下：

原始 DataFrame：
   姓名  年龄 性别
0  张三   25  男
1  李四   30  女
2  王五   28  男

添加序号后的 DataFrame：
   序号  姓名  年龄 性别
0   1  张三   25  男
1   2  李四   30  女
2   3  王五   28  男

如上所示，序号列成功添加，并且每个姓名对应的序号与其在 DataFrame 中的位置一致。

结论

本文介绍了如何使用 pandas 库为 DataFrame 添加一列序号。这种操作在数据分析中非常常见，有助于后续的数据处理与分析。希望通过今天的分享，能够帮助您更好地掌握 pandas 库的基本操作。无论是在数据清洗、分析还是可视化中，为数据添加序号列都是至关重要的步骤。欢迎您在实际项目中试验并应用这些技巧！