在Python的pandas库中,你可以通过多种方法在DataFrame中添加新列。以下是两个常见的方法:
方法一:直接赋值
python
import pandas as pd
创建一个示例DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
添加一个新列C,初始值为0
df['C'] = 0
print(df)
在这个例子中,新列C的初始值全部为0。
方法二:使用apply函数
python
import pandas as pd
创建一个示例DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
定义一个函数,将应用到DataFrame的每一行,将新列的值设为A列值和B列值之和
def add_column(row):
return row['A'] + row['B']
使用apply函数添加新列C
df['C'] = df.apply(add_column, axis=1)
print(df)
在这个例子中,新列C的值是A列值和B列值之和。注意,这个方法通常在你需要在每行计算出一个新列值的情况下使用。