在熊猫python中过滤日期时间列

Question

我正在尝试动态地datetime column子集，但仍将column保留在dataframe中，因为该函数将用于UI，但是由于某些原因，我应用该功能后看不到该列。

这是执行此操作的辅助功能

def get_df(df, start, end, datetime_col, columns):
    df = df.loc[(df[datetime_col]>=start) & (df[datetime_col]<= end)][columns]
    return df

这是我应用该功能的方式

get_df(df, start=datetime(2020,3,1), end=default_end_date, datetime_col="time_iso8601", columns=["Berlin"]).head()


   Berlin
0   0
1   6
2   9
3   19
4   24

我希望将datetime_col包含在数据框中，因为它原来是

Answer 1

这应该有效

def get_df(df, start, end, datetime_col, columns):
        df = df.loc[(df[datetime_col]>=start) & (df[datetime_col]<= end)][[columns,datetime_col]]
        return df