Python|熊猫 dataframe.insert()

熊猫是一个强大的数据处理工具，它提供了众多数据处理和操作方法，其中 dataframe.insert() 是其中之一。

定义

dataframe.insert() 方法用于在指定列索引之前插入新的一列。语法如下:

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

参数说明：

• loc：要插入的列索引位置（类型为 int）。如果为 0，则在最前面插入。如果为 len(dataframe.columns)，则在最后面插入。
• column：用于插入数据的新列的名称。
• value：新列的数据。
• allow_duplicates：如果新列名已经存在于 dataframe 中，allow_duplicates 控制是否添加此列。

示例

以下示例将插入一列名为 'new_col' 的新列，其值为 1，将插入到 dataframe 中的第二个位置。

import pandas as pd

dataframe = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
dataframe.insert(1, 'new_col', 1)

以下示例将在 dataframe 中添加与列 B 相同的新列，并将其插入到列 B 之前。

import pandas as pd

dataframe = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
dataframe.insert(dataframe.columns.get_loc('B'), 'B_copy', dataframe['B'])

注意事项

• 如果新列的数据不是当前 dataframe 与旧列的匹配，它不会直接替换原有列。因此，在某些情况下，需要使用 .drop() 方法来删除原有列。
• 当 allow_duplicates=True 时，允许添加名为 'column' 的列多次，这会在列名后面添加数字，例如 'column_1'。