pandas简单实现excel中countif函数

原作者问题：pandas如何简单实现excel中的countif函数？

题主有如上所示的一个DataFrame，需要实现excel中countif函数，即统计每一行中除了指标之外的x1, x2, x3, x4中大于指标的个数。如第一行“4，5，5，3”中有3个数大于指标3，那么count就应该是3。

---

本质上来讲，excel中大部分函数都是通过对一行或一列的计算（也有对一块计算，另说）然后把相同的计算映射到整个表中。在python里apply函数就很好地解决了这个问题。下面是这个问题的解决代码。

import pandas as pd

base = [3,3,4,3,1,3,3,3,5,2,3,3,3,3]
x_1 = [4,4,4,4,2,1,4,3,4,3,4,4,2,4]
x_2 = [5,3,4,4,2,4,5,1,4,4,1,3,5,1]
x_3 = [5,2,4,1,1,5,4,3,2,3,3,4,2,4]
x_4 = [3,2,3,5,1,3,3,5,4,2,5,5,5,5]

df = pd.DataFrame({'指标':base, 'X1':x_1, 'X2':x_2, 'X3':x_3, 'X4':x_4})
#print(df)

############### 上面为生成图示数据 #####################

#count_if函数
def countif(line, base, count_column):
	line['cnt_if'] = sum(line[count_column] > line[base])
	return line

#将count_if应用到每一行
new_df = df.apply(countif, axis=1, args=('指标',['X1', 'X2', 'X3', 'X4'], ))
print(new_df)

countif函数就是模拟了每一行的操作，把每一line中需要统计的列大于base列个数计算出来，结果放到新列’cnt_if’中，再用apply映射到整个表里。

结果如下：

以上。