根据pandas中的if条件删除数据帧列

时间:2017-07-26 13:35:24

标签: python pandas

我有以下数据框

 2001-01-01   2001-01-02            2001-01-03 
   1               0                   8 

我想删除每一列,小于2001-01-02,即我的df应如下所示:

  2001-01-02            2001-01-03
          0                   8 

有人知道怎么做吗?

2 个答案:

答案 0 :(得分:4)

使用&#39;反向&#39;条件<>=,因为只需要等于或高于的值:

df = pd.DataFrame([[1,0,8]], columns = pd.date_range('2001-01-01', periods=3))
print (df)
   2001-01-01  2001-01-02  2001-01-03
0           1           0           8

print (df.columns >= '2001-01-02')
[False  True  True]


df1 = df.loc[:, df.columns >= '2001-01-02']
print (df1)
   2001-01-02  2001-01-03
0           0           8

cols = df.columns[df.columns >= '2001-01-02']
df1 = df[cols]
print (df1)
   2001-01-02  2001-01-03
0           0           8

另一个解决方案是为~添加boolean array

df1 = df.loc[:, ~(df.columns < '2001-01-02')]
print (df1)
   2001-01-02  2001-01-03
0           0           8

答案 1 :(得分:0)

您可以使用label slices with loc

df.loc[:, :'2001-01-02']

   2001-01-01  2001-01-02
0           1           0

df.loc[:, '2001-01-02':]

   2001-01-02  2001-01-03
0           0           8