时间戳

首页 > 解决方案 > 如何找到三列中的数字超过熊猫中其他列值的一半

python - 如何找到三列中的数字超过熊猫中其他列值的一半

问题描述

X、Y 和 Z 三列具有随机值。

X、Y 和 Z 值不应小于彼此值的一半。

在数学逻辑方面需要帮助

  X    Y     Z
5461  919  8269
5400  5000 7000  <---
4000   1     35

输出应该是 5400 5000 7000。在第一行，Y 小于 X 和 Z 的一半。在第三行，Y 和 Z 都小于 X 的一半。

标签： pythonpandasmath

解决方案

逻辑很简单：只要检查每个值是否小于其他两个值的一半。使用 apply 返回True或False然后使用获得的boolean值过滤行。

import pandas as pd
a = pd.DataFrame([[
5461 , 919  ,8269],
[5400 , 5000, 7000], 
[4000 ,  1  ,   35]], columns = ['X','Y','Z'])
print(a)

def less(a,b,c):
    if a<b/2 or a<c/2:
        return True

def cond(a,b,c):
    if less(a,b,c) or less(b,a,c) or less(c,a,b):
        return False
    return True

a[a.apply(lambda x: cond(x['X'],x['Y'],x['Z']), axis=1)]

输出

      X     Y     Z
1  5400  5000  7000

推荐阅读