如何计算DataFrame列中符合条件的前n个元素的和?
解决方法
要计算DataFrame第一列中前3个负数的和,你需要先筛选出该列的负数元素,再取前3个进行求和,而非直接对所有负数求和。具体代码如下:
假设你的DataFrame名为df,可以分步实现:
# 提取第一列数据 first_col = df.iloc[:, 0] # 筛选负数、取前3个后求和 sum_top3_neg = first_col[first_col < 0].head(3).sum()
或者合并为一行代码:
sum_top3_neg = df.iloc[:, 0][df.iloc[:, 0] < 0].head(3).sum()
问题原因说明
你之前用loc或iloc时,应该是直接筛选了该列所有负数元素并求和,没有通过head(3)限制只取前3个符合条件的元素,因此得到的是所有负数的总和,而非目标的前3个负数之和。
内容的提问来源于stack exchange,提问作者user55848
相关产品推荐
相关产品推荐

