pandas.DataFrame删除/选取含有特定数值的行或列

1.删除/选取某列含有特殊数值的行

import pandas as pd
import numpy as np

a=np.array([[1,2,3],[4,5,6],[7,8,9]])
df1=pd.DataFrame(a,index=[row0,row1,row2],columns=list(ABC))
print(df1)
df2=df1.copy()

#删除/选取某列含有特定数值的行
#df1=df1[df1[A].isin([1])]
#df1[df1[A].isin([1])]  选取df1中A列包含数字1的行

df1=df1[~df1[A].isin([1])]
#通过~取反,选取不包含数字1的行
print(df1)

运行结果:

2.删除/选取某行含有特殊数值的列

 

#删除/选取某行含有特定数值的列
cols=[x for i,x in enumerate(df2.columns) if df2.iat[0,i]==3]
#利用enumerate对row0进行遍历,将含有数字3的列放入cols中
print(cols)

#df2=df2[cols]   选取含有特定数值的列
df2=df2.drop(cols,axis=1) #利用drop方法将含有特定数值的列删除
print(df2)

运行结果:

3.删除含有空值的行或列

实现思路:利用对空值赋予特定值,再利用上文介绍的方法找到这些含有特定值的行或列去除即可。

import pandas as pd
import numpy as np

df1 = pd.DataFrame(
    [
        [np.nan, 2, np.nan, 0],
        [3, 4, np.nan, 1],
        [np.nan, np.nan, np.nan, 5],
        [np.nan, 3, np.nan, 4]
    ],columns=list(ABCD))
print(df1)
df2=df1.copy()

df1[A]=df1[A].fillna(null) #将df中A列所有空值赋值为null
print(df1)
df1=df1[~df1[A].isin([null])]
print(df1)

#删除某行空值所在列

df2[0:1]=df2[0:1].fillna(null)
print(df2)
cols=[x for i,x in enumerate(df2.columns) if df2.iat[0,i]==null]
print(cols)
df2=df2.drop(cols,axis=1)
print(df2)

运行结果:

经验分享 程序员 微信小程序 职场和发展