从Python函数调用`dt.`类型

Question

我在pandas中有时间戳，我希望能够编写一个函数，使我能够提取日期的不同部分。这是一个MWE：

import pandas as pd
df = pd.DataFrame({'hit': pd.to_datetime(['2018-12-29 00:00:00', '2019-03-25 00:00:00'])})

这就是我想做的：

def test(timepart):
    df['hit'].dt.timepart

所以我可以调用test(month)，函数将返回12, 03

Answer 1

您可以使用strftime：

def test(timepart):
    return df['hit'].dt.strftime('%'+timepart[0])
print(test('month'))

输出：

0    12
1    03
Name: hit, dtype: object