python - 过滤熊猫python中的日期时间列

Question

我正在尝试datetime column动态地对 a 进行子集化，但仍将其保留column在其中，dataframe因为该函数将用于 aUI但由于某些原因，在应用该函数后我看不到该列。

这是执行此操作的辅助函数

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原样包含在数据框中

标签： pythonpython-3.xpandasdataframe

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