使用numpy库中的广播机制来为DataFrame的列赋值。广播机制可以将一维数组扩展为和DataFrame相同的维度,在进行计算时自动进行对齐操作。
代码示例:
import pandas as pd import numpy as np
df = pd.DataFrame({'A': [1,2,3], 'B': [4,5,6]})
df['C'] = np.array([7,8,9])
print(df)
输出结果: A B C 0 1 4 7 1 2 5 8 2 3 6 9
上一篇:不使用循环填充矩阵的R代码