我想删除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
答案 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)