在Pandas中添加新列,可以使用assign()函数添加列,并使用numpy.arange()函数生成数字范围序列。
示例代码如下:
import pandas as pd
import numpy as np
# 创建数据框
df = pd.DataFrame({'A': ['a', 'b', 'c', 'd'], 'B': ['e', 'f', 'g', 'h']})
# 添加新列
df = df.assign(C=np.arange(1, len(df) + 1))
# 打印结果
print(df)
该代码将在现有数据框中添加一个名为“C”的新列,并使用numpy.arange函数生成数字范围序列。结果输出如下:
A B C
0 a e 1
1 b f 2
2 c g 3
3 d h 4
其中,“C”列的值为1到4,表示数据框中的行数。