删除pandas数据帧中全部为零的行

时间:2014-03-26 01:20:10

标签: python pandas

我可以使用pandas dropna()功能删除部分或全部列设置为NA的行。是否有一个等效函数用于删除所有列值为0的行?

P   kt  b   tt  mky depth
1   0   0   0   0   0
2   0   0   0   0   0
3   0   0   0   0   0
4   0   0   0   0   0
5   1.1 3   4.5 2.3 9.0

在这个例子中,我们想从数据框中删除前4行。

谢谢!

14 个答案:

答案 0 :(得分:82)

一衬垫。不需要转置:

df.loc[~(df==0).all(axis=1)]

对于那些喜欢对称的人来说,这也有效......

df.loc[(df!=0).any(axis=1)]

答案 1 :(得分:78)

事实证明,这可以用矢量化方式很好地表达:

> df = pd.DataFrame({'a':[0,0,1,1], 'b':[0,1,0,1]})
> df = df[(df.T != 0).any()]
> df
   a  b
1  0  1
2  1  0
3  1  1

答案 2 :(得分:16)

我每个月都会查询一次这个问题,并且总是要从评论中找出最佳答案:

-Xfatal-warnings

谢谢Dan Allan!

答案 3 :(得分:6)

用nan替换零,然后删除所有条目为nan的行。 之后用零替换nan。

df= df.replace(0,nan)
df=df.dropna(how='all',axis=0)
df=df.replace(nan,0)

答案 4 :(得分:4)

我认为这种解决方案是最短的:

df= df[df['ColName'] != 0]

答案 5 :(得分:3)

import pandas as pd

df = pd.DataFrame({'a' : [0,0,1], 'b' : [0,0,-1]})

temp = df.abs().sum(axis=1) == 0      
df = df.drop(temp)

结果:

>>> df
   a  b
2  1 -1

答案 6 :(得分:3)

我发现在解决这个问题时我发现了一些有用的解决方案,特别是对于更大的数据集:

df[(df.sum(axis=1) != 0)]       # 30% faster 
df[df.values.sum(axis=1) != 0]  # 3X faster 

继续@ U2EF1的例子:

In [88]: df = pd.DataFrame({'a':[0,0,1,1], 'b':[0,1,0,1]})

In [91]: %timeit df[(df.T != 0).any()]
1000 loops, best of 3: 686 µs per loop

In [92]: df[(df.sum(axis=1) != 0)]
Out[92]: 
   a  b
1  0  1
2  1  0
3  1  1

In [95]: %timeit df[(df.sum(axis=1) != 0)]
1000 loops, best of 3: 495 µs per loop

In [96]: %timeit df[df.values.sum(axis=1) != 0]
1000 loops, best of 3: 217 µs per loop

在更大的数据集上:

In [119]: bdf = pd.DataFrame(np.random.randint(0,2,size=(10000,4)))

In [120]: %timeit bdf[(bdf.T != 0).any()]
1000 loops, best of 3: 1.63 ms per loop

In [121]: %timeit bdf[(bdf.sum(axis=1) != 0)]
1000 loops, best of 3: 1.09 ms per loop

In [122]: %timeit bdf[bdf.values.sum(axis=1) != 0]
1000 loops, best of 3: 517 µs per loop

答案 7 :(得分:2)

您可以使用快速lambda函数检查给定行中的所有值是否为0。然后,您可以使用应用lambda的结果来选择仅匹配或不匹配该条件的行:

import pandas as pd
import numpy as np

np.random.seed(0)

df = pd.DataFrame(np.random.randn(5,3), 
                  index=['one', 'two', 'three', 'four', 'five'],
                  columns=list('abc'))

df.loc[['one', 'three']] = 0

print df
print df.loc[~df.apply(lambda row: (row==0).all(), axis=1)]

收率:

              a         b         c
one    0.000000  0.000000  0.000000
two    2.240893  1.867558 -0.977278
three  0.000000  0.000000  0.000000
four   0.410599  0.144044  1.454274
five   0.761038  0.121675  0.443863

[5 rows x 3 columns]
             a         b         c
two   2.240893  1.867558 -0.977278
four  0.410599  0.144044  1.454274
five  0.761038  0.121675  0.443863

[3 rows x 3 columns]

答案 8 :(得分:1)

另一种选择:

# Is there anything in this row non-zero?
# df != 0 --> which entries are non-zero? T/F
# (df != 0).any(axis=1) --> are there 'any' entries non-zero row-wise? T/F of rows that return true to this statement.
# df.loc[all_zero_mask,:] --> mask your rows to only show the rows which contained a non-zero entry.
# df.shape to confirm a subset.

all_zero_mask=(df != 0).any(axis=1) # Is there anything in this row non-zero?
df.loc[all_zero_mask,:].shape

答案 9 :(得分:1)

这对我有用 <div class="bottom-sticky-ad closeable"> <div class="ad-wrapper touch-display-none"> </div> </div>

答案 10 :(得分:0)

df = df [~( df [ ['kt'  'b'   'tt'  'mky' 'depth', ] ] == 0).all(axis=1) ]

尝试使用此命令即可正常运行。

答案 11 :(得分:0)

对我来说,这段代码:df.loc[(df!=0).any(axis=0)] 不工作。它返回了确切的数据集。

相反,我使用df.loc[:, (df!=0).any(axis=0)]并删除了数据集中所有具有0值的列

函数.all()删除了我数据集中所有零值的列。

答案 12 :(得分:0)

按照 accepted answer 中的示例,一个更优雅的解决方案:

df = pd.DataFrame({'a':[0,0,1,1], 'b':[0,1,0,1]})
df = df[df.any(axis=1)]
print(df)

   a  b
1  0  1
2  1  0
3  1  1

答案 13 :(得分:-2)

要在任何行中删除所有值为0的列:

new_df = df[df.loc[:]!=0].dropna()