pandas insert() 插入列

看过来

《pandas 教程》持续更新中，提供建议、纠错、催更等加作者微信: gairuo123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。跟作者学习，请进入 Python学习课程。欢迎关注作者出版的书籍：《深入浅出Pandas》和《Python之光》。

pandas 中的 insert() 方法用于在 DataFrame 的指定位置插入一列。它允许用户在 DataFrame 的特定列之前插入新的数据列，而不覆盖现有的列。

功能

功能特点：

指定位置插入：可以通过指定索引位置来插入新列。
标签和数据：新列需要指定标签（列名）和数据（可以是列表、数组或 Series）。
可选填充值：如果不提供数据，可以使用填充值。

语法：

DataFrame.insert(loc, column, value, allow_duplicates=False)

参考：

loc：插入位置的索引（整数）。
column：新列的标签（列名）。
value：新列的数据（可以是列表、数组、Series 或标量）。
allow_duplicates：是否允许重复列名，默认为 False。

支持的对象：

pandas.DataFrame.insert
pandas.Index.insert
pandas.api.extensions.ExtensionArray.insert

注意事项：

索引范围：插入位置 loc 的索引值必须在 0 到 len(columns) 之间。
重复列名：默认情况下，insert() 不允许重复列名。如果需要允许重复列名，可以将 allow_duplicates 参数设置为 True。
数据对齐：新列的数据长度必须与 DataFrame 的行数匹配。

示例

以下是一个完整的示例，展示了如何使用 insert() 方法在 DataFrame 的不同位置插入新列：

import pandas as pd

# 创建一个示例 DataFrame
data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6]
}
df = pd.DataFrame(data)

# 在第 1 列位置插入新列 'C'
new_column_data = [7, 8, 9]
df.insert(1, 'C', new_column_data)
print('插入新列 C 后的 DataFrame:')
print(df)

# 在第 0 列位置插入新列 'D'
new_column_data = [10, 11, 12]
df.insert(0, 'D', new_column_data)
print('插入新列 D 后的 DataFrame:')
print(df)
'''
插入新列 C 后的 DataFrame:
   A  C  B
0  1  7  4
1  2  8  5
2  3  9  6

插入新列 D 后的 DataFrame:
    D  A  C  B
0  10  1  7  4
1  11  2  8  5
2  12  3  9  6
'''

通过 insert() 方法，可以灵活地在 DataFrame 的指定位置插入新列，从而更方便地进行数据操作和管理。

案例

案例如下：

>>> df = pd.DataFrame({'col1': [1, 2], 'col2': [3, 4]})
>>> df
   col1  col2
0     1     3
1     2     4
>>> df.insert(1, "newcol", [99, 99])
>>> df
   col1  newcol  col2
0     1      99     3
1     2      99     4
>>> df.insert(0, "col1", [100, 100], allow_duplicates=True)
>>> df
   col1  col1  newcol  col2
0   100     1      99     3
1   100     2      99     4

请注意，pandas在值来自Series类型的情况下使用索引对齐：

>>> df.insert(0, "col0", pd.Series([5, 6], index=[1, 2]))
>>> df
   col0  col1  col1  newcol  col2
0   NaN   100     1      99     3
1   5.0   100     2      99     4

Index

Index.insert(loc, item) 创建新索引，在位置插入新项目。例如：

>>> idx = pd.Index(['a', 'b', 'c'])
>>> idx.insert(1, 'x')
Index(['a', 'x', 'b', 'c'], dtype='object')

ExtensionArray

pandas.api.extensions.ExtensionArray.insert 的语法是 ExtensionArray.insert(loc, item) 在给定位置插入一个项目。如：

>>> arr = pd.array([1, 2, 3])
>>> arr.insert(2, -1)
<IntegerArray>
[1, 2, -1, 3]
Length: 4, dtype: Int64

参考

https://pandas.pydata.org/docs/reference/api/pandas.Index.insert.html
https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.insert.html
https://pandas.pydata.org/docs/reference/api/pandas.api.extensions.ExtensionArray.insert.html

< squeeze() 数据挤压降维 pandas 函数详细介绍 pop() 删除并返回指定列 >

< assign() 指定新列 pandas 添加修改数据 eval() 表达式计算 >

更新时间：2024-07-13 20:32:58 标签：pandas python 插入