温馨提示×

Pandas中怎么使用广播机制

小亿
82
2024-05-11 16:56:51
栏目: 编程语言

在Pandas中,可以使用广播机制来对两个不同形状的DataFrame或Series进行操作。广播机制会自动将大小不同的DataFrame或Series进行扩展,使其具有相同的形状,然后再进行运算。

比如,假设有一个DataFrame df1 和一个 Series s1,我们想要将s1加到df1的每一行上,可以使用广播机制来实现:

import pandas as pd

# 创建一个DataFrame
df1 = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})

# 创建一个Series
s1 = pd.Series([10, 20])

# 使用广播机制将s1加到df1的每一行上
result = df1 + s1

print(result)

输出结果为:

     A     B
0  11.0  14.0
1  22.0  25.0
2   NaN   NaN

可以看到,s1被自动扩展成了与df1相同的形状,然后进行了加法运算。需要注意的是,如果两个对象在某些位置上无法对齐,那么广播机制会在这些位置上产生缺失值(NaN)。

0