如何在未知列长度时使用pandas.insert()插入全1列？

使用insert函数的实现方式 #

要解决不知道需要多少个1的问题，核心是先获取DataFrame的行数，再构造对应长度的全1序列：

1. 获取DataFrame的总行数：可以用df.shape[0]或者len(df)，两者都能得到准确的行数。
2. 构造全1的序列：推荐用pd.Series(1, index=df.index)，这样能保证索引和原DataFrame完全匹配，避免因索引不兼容出现的错误；也可以用[1] * row_num构造列表，但前者更稳妥。
3. 调用insert函数插入列：指定插入位置、列名和值即可。

示例代码：

import pandas as pd

# 示例DataFrame
df = pd.DataFrame({'A': [10, 20, 30], 'B': [40, 50, 60]})

# 获取行数
row_num = df.shape[0]
# 构造匹配索引的全1序列
ones_col = pd.Series(1, index=df.index)
# 插入到第0列（最左侧），列名为'ones'
df.insert(loc=0, column='ones', value=ones_col)

更简便的替代方法 #

其实不用手动计算行数，利用pandas的广播机制可以更高效实现：

• 直接赋值添加列：这是最快捷的方式，pandas会自动将单个值扩展为与DataFrame行数一致的列：

df['ones'] = 1

• 使用assign方法：适合需要保留原DataFrame的场景，该方法返回添加新列后的新DataFrame，不会修改原数据：

new_df = df.assign(ones=1)

内容的提问来源于stack exchange，提问作者Sergio