使用pandas实现excel中countif函数

在pandas中实现类似于excel countif函数的功能非常简单。假如有如下所示的数据集，其中需要统计每行中大于“指标”值的列的个数：

import pandas as pd

data = pd.dataframe({
    'x1': [10, 20, 30, 40],
    'x2': [15, 25, 35, 45],
    'x3': [20, 30, 40, 50],
    'x4': [25, 35, 45, 55],
    '指标': [18, 28, 38, 48]
})

可以通过以下代码实现此功能：

data['countif'] = data.apply(lambda x : sum(data.loc[x.name, 'X1':'X4'] > data.loc[x.name, '指标']), axis =1)

其中：

• apply(lambda x : ...)：将自定义函数应用于每一行数据。
• x.name：获取当前行的索引值。
• data.loc[x.name, 'x1':'x4']：获取当前行的'x1'到'x4'列数据。
• > data.loc[x.name, '指标']：将当前行的'x1':'x4'列与'指标'列比较，结果为布尔值。
• sum(...)：统计比较结果中true的个数。