原作者问题: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映射到整个表里。

结果如下:

以上。

更多推荐

pandas简单实现excel中countif函数