python - 将函数应用于数据框

Question

我刚开始使用 apply:

import pandas
df=pandas.DataFrame({"k1":['2017-03-07','2017-02-07','2017-01-15'],
                 "k2":[1,2,3],})

我想获取匹配某个月份的数据。例如:我想获取月份==2:

df[pandas.to_datetime(df.k1).dt.month == 2]

但是我想写一个参数可以匹配月份的函数。这是我的职责。这不起作用。

def get_month(df, mon=1):
    return df[pandas.to_datetime(df.k1).dt.month == mon]

df.apply(get_month)

Answer 1

您需要稍微重构一下您的代码。 apply 每行有效。

代码:

def get_month(mon):
    return lambda x: pandas.to_datetime(x.k1).month == mon

测试代码:

df = pandas.DataFrame({"k1": ['2017-03-07', '2017-02-07', '2017-01-15'],
                       "k2": [1, 2, 3], })

print(df[df.apply(get_month(1), axis=1)])

结果:

           k1  k2
2  2017-01-15   3