我想在python中删除一行

时间:2017-04-04 23:35:23

标签: python dataframe

我想删除python中列[3]不为0的行。

#prepare
from numpy import nan as NA
from pandas import Series,DataFrame

data=DataFrame([[1.,6.5,3.,0],[1.,NA,NA,0.],[NA,NA,NA,1],[NA,6.5,3.,7.]])
data
     0    1    2    3
0  1.0  6.5  3.0  0.0
1  1.0  NaN  NaN  0.0
2  NaN  NaN  NaN  1.0
3  NaN  6.5  3.0  7.0

我执行了以下代码,但效果不佳。

Third=data[3]
for i in range(len(data.index)):
if not Third[i] is 0:
    drop_idx=[i]
    data.drop(drop_idx,axis=0)

结果

     0    1    2    3
1  1.0  NaN  NaN  0.0
2  NaN  NaN  NaN  1.0
3  NaN  6.5  3.0  7.0
     0    1    2    3
0  1.0  6.5  3.0  0.0
2  NaN  NaN  NaN  1.0
3  NaN  6.5  3.0  7.0
     0    1    2    3
0  1.0  6.5  3.0  0.0
1  1.0  NaN  NaN  0.0
3  NaN  6.5  3.0  7.0
     0    1    2    3
0  1.0  6.5  3.0  0.0
1  1.0  NaN  NaN  0.0
2  NaN  NaN  NaN  1.0

我想知道如何得到以下结果。你能告诉我哪里错了吗?

     0    1    2    3
0  1.0  6.5  3.0  0.0
1  1.0  NaN  NaN  0.0

1 个答案:

答案 0 :(得分:0)

这将完成这项工作:

import pandas as pd
data=pd.DataFrame([[1.,6.5,3.,0],[1.,None,None,0.],[None,None,None,1],[None,6.5,3.,7.]])
data=data[data[3]==0]

然后data

    0   1   2   3
0   1.0 6.5 3.0 0.0
1   1.0 NaN NaN 0.0

你也可以按照以下方式进行:

import pandas as pd
data=pd.DataFrame([[1.,6.5,3.,0],[1.,None,None,0.],[None,None,None,1],[None,6.5,3.,7.]])
data.drop(data[data[3]!=0].index,inplace=True)