根据熊猫中的 if 条件删除数据框列

fre*_*888 3 python pandas

我有以下数据框

 2001-01-01   2001-01-02            2001-01-03 
   1               0                   8 
Run Code Online (Sandbox Code Playgroud)

我想删除小于 2001-01-02 的每一列,即我的 df 应该是这样的:

  2001-01-02            2001-01-03
          0                   8 
Run Code Online (Sandbox Code Playgroud)

有人知道怎么做吗?

jez*_*ael 5

使用 'inverse' 条件<>=,因为只需要相等或更高的值:

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
Run Code Online (Sandbox Code Playgroud)

另一种解决方案是添加~inverse boolean array

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