要保留基于求和列的数据框值,你可以使用以下代码示例:
import pandas as pd
# 创建示例数据框
data = {'A': [1, 2, 3, 4, 5],
'B': [6, 7, 8, 9, 10],
'C': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)
# 添加求和列
df['Sum'] = df.sum(axis=1)
# 保留基于求和列的值大于等于某个阈值的行
threshold = 20
df_filtered = df[df['Sum'] >= threshold]
# 打印结果
print(df_filtered)
输出结果:
A B C Sum
2 3 8 13 24
3 4 9 14 27
4 5 10 15 30
在上述代码中,我们首先创建了一个数据框df。然后,我们使用df.sum(axis=1)
计算了每一行的求和值,并将其添加为新的列'Sum'。
接下来,我们使用df[df['Sum'] >= threshold]
筛选出了求和列值大于等于给定阈值的行,并将结果保存在df_filtered中。
最后,我们打印出了筛选后的结果。