我正在翻译熊猫的Excel公式。对具有指定条件的列进行计数并逐行累加。如果所选列中的某个单元格满足给定条件,则必须按行进行计数,然后添加满足条件的计数。
我有数据框:
df:
a b c
14 x1 2
17 x2 2
0 x,1 3
1 x1 1
Excel公式:
= COUNTIFS($U2,14,$X2,"x2",$W2,2)+COUNTIFS($U2,17,$X2,"x2",$W2,2)+COUNTIFS(U2,14,$X2,"x1",$W2,2)
熊猫公式:
df['counted'] = (df[(df['a']==14) & (df['b']=='x2') & (df['c']==2)].count(axis=1)) + (df[(df['a']==17) & (df['b']=='x2') & (df['c']==2)].count(axis=1)) + (df[(df['a']==14) & (df['b']=='x1') & (df['c']==2)].count(axis=1))
我从熊猫公式中得到以下结果:df:
a b c counted
14 x1 2 NaN
17 x2 2 NaN
0 x,1 3 NaN
1 x1 1 NaN
预期结果如下所示。任何获得正确配方的帮助将不胜感激。
预期结果df:
a b c counted
14 x1 2 0
17 x2 2 1
0 x,1 3 0
1 x1 1 0
我相信您需要将sum
布尔值掩码转换为整数:
a = (df['a']==14) & (df['b']=='x2') & (df['c']==2)
b = (df['a']==17) & (df['b']=='x2') & (df['c']==2)
c = (df['a']==14) & (df['b']=='x1') & (df['c']==2)
避免重复以获得更好的性能也是可能的连锁条件:
m1 = df['a']==14
m2 = df['b']=='x2'
m3 = df['c']==2
m4 = df['a']==17
m5 = df['b']=='x1'
a = m1 & m2 & m3
b = m4 & m2 & m3
c = m1 & m5 & m3
df['counted'] = a.astype(int)+ b.astype(int) + c.astype(int)
print (df)
a b c counted
0 14 x1 2 1
1 17 x2 2 1
2 0 x,1 3 0
3 1 x1 1 0
或按位链掩码OR
,然后转换为整数:
df['counted'] = (a | b | c).astype(int)
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句