根据B列的特定范围计算A列的avg

Question

我有一个2列的数据格式。速度(A)和海拔(B)。我想根据(B)栏中的范围求出(A)栏的平均值

例如:我希望选择(B)中值在-0.5到+0.8之间的所有行，并获得(A)中的平均速度。

有人能帮忙吗？

Answer 1

如果只需要一个标量来表示列'A'的平均值，其中某些基于列'B'的行是匹配的，则可以使用.loc，您可以阅读更多关于这里的内容。

例如：

df = pd.DataFrame({'A': [1,2,3,4,5], 'B' : [-1,0,0.1,0.2,1]})

>>> df
   A    B
0  1 -1.0
1  2  0.0
2  3  0.1
3  4  0.2
4  5  1.0

然后，下面一行用B的某个条件计算A列子集的平均值：

>>> df.loc[df['B'].between(-0.5,0.8) , 'A'].mean()
3.0