在循环中向DataFrame添加新列,可以使用DataFrame的assign()
方法或直接给DataFrame赋值的方式。
使用assign()
方法时,可以通过lambda函数或其他函数来计算新列的值,并将其赋给DataFrame的新列。例如:
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# 使用assign()方法添加新列
df = df.assign(C=lambda x: x['A'] + x['B'])
print(df)
输出结果为:
A B C
0 1 4 5
1 2 5 7
2 3 6 9
另一种方式是直接给DataFrame赋值,通过索引操作来添加新列。例如:
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# 直接给DataFrame赋值添加新列
df['C'] = df['A'] + df['B']
print(df)
输出结果为:
A B C
0 1 4 5
1 2 5 7
2 3 6 9
以上两种方法都可以在循环中使用,根据具体需求在每次循环迭代时添加新列。
领取专属 10元无门槛券
手把手带您无忧上云