Question

嘿伙计们我想从我的熊猫DataFrame中删除少于3个非零值（不包括总列数）的行。

所以现在我有。

    year    2001 2002 2003 2004 2005 2006 2007 TOTAL
    player  
    Emma    0     0     0    0    3    4    5    12
    Max     3     5     0    0    0    0    0    8
    Josh    1     2     4    1    2    1    0    11
    Steve   0     0     0    0    3    0    0    3
    Mike    1     0     0    0    0    0    2    3

但我想：

    year    2001 2002 2003 2004 2005 2006 2007 TOTAL
    player  
    Emma    0     0     0    0    3    4    5    12
    Josh    1     2     4    1    2    1    0    11

我正在考虑使用for循环，但我不确定如何实现它/如果它是解决问题的最佳方法。

Answer 1

<强> pandas
我drop TOTAl和sum每行的非零数

df[df.drop('TOTAL', 1).ne(0).sum(1).gt(2)]

year    2001  2002  2003  2004  2005  2006  2007  TOTAL
player                                                 
Emma       0     0     0     0     3     4     5     12
Josh       1     2     4     1     2     1     0     11

<强> numpy
更快的解决方案

v = df.values
m = (v[:, :-1] != 0).sum(1) > 2
pd.DataFrame(v[m], df.index[m], df.columns)

year    2001  2002  2003  2004  2005  2006  2007  TOTAL
player                                                 
Emma       0     0     0     0     3     4     5     12
Josh       1     2     4     1     2     1     0     11

Answer 2

<强>设置

df = pd.DataFrame({'2001': {'Emma': 0, 'Josh': 1, 'Max': 3, 'Mike': 1, 'Steve': 0},
 '2002': {'Emma': 0, 'Josh': 2, 'Max': 5, 'Mike': 0, 'Steve': 0},
 '2003': {'Emma': 0, 'Josh': 4, 'Max': 0, 'Mike': 0, 'Steve': 0},
 '2004': {'Emma': 0, 'Josh': 1, 'Max': 0, 'Mike': 0, 'Steve': 0},
 '2005': {'Emma': 3, 'Josh': 2, 'Max': 0, 'Mike': 0, 'Steve': 3},
 '2006': {'Emma': 4, 'Josh': 1, 'Max': 0, 'Mike': 0, 'Steve': 0},
 '2007': {'Emma': 5, 'Josh': 0, 'Max': 0, 'Mike': 2, 'Steve': 0},
 'TOTAL': {'Emma': 12, 'Josh': 11, 'Max': 8, 'Mike': 3, 'Steve': 3}})

<强>解决方案

df.loc[np.sum(df.iloc[:,:-1]>0, axis=1)[lambda x: x>=3].index]
    Out[889]: 
      2001  2002  2003  2004  2005  2006  2007  TOTAL
Emma     0     0     0     0     3     4     5     12
Josh     1     2     4     1     2     1     0     11

或者使用groupby和filter：

df.groupby(level=0).filter(lambda x: np.sum(x.iloc[0,:]>0)>=4)
Out[918]: 
      2001  2002  2003  2004  2005  2006  2007  TOTAL
Emma     0     0     0     0     3     4     5     12
Josh     1     2     4     1     2     1     0     11

删除熊猫中少于3个非零值的行

2 个答案: