代码之家 › 专栏 › 技术社区 › Tim

np.vectorize和relativelta返回“relativeltaonlydiffs datetime/date”

vectorization date pandas python

Tim · 技术社区 · 2 年前

我有一个pandas数据帧,它有两个datetime64[ns]列(“d1”和“d2”)表示日期。我想创建第三列,计算这两个日期之间的差异。我不能使用简单的365天计算,所以我需要relativelta。

在一行中使用relativelta效果良好:

import dateutil.relativedelta as relativedelta
relativedelta.relativedelta(df["d1"][0],df["d2"][0])
> relativedelta(years=+1)

但它在列上失败了。所以我将其矢量化:

date_diffs=np.vectorize(relativedelta.relativedelta)

然后我试着

date_diffs(df["d1"],df["d2"])

但这又回来了 TypeError: relativedelta only diffs datetime/date

我该如何解决这个问题?或者我应该简单地使用 apply 语句还是for循环?

1 回复 | 直到 2 年前

jezrael 2 年前

使用列表理解:

df = pd.DataFrame({'d1':pd.date_range('2000-01-05', periods=3),
                   'd2':pd.date_range('2006-08-05', periods=3, freq='35M')})

from dateutil.relativedelta import relativedelta

def date_diffs(s, e):
    return relativedelta(s,e)

df['out'] = [date_diffs(s, e) for s, e in zip(df["d1"],df["d2"])]
print(df)
          d1         d2                                            out
0 2000-01-05 2006-08-31   relativedelta(years=-6, months=-7, days=-26)
1 2000-01-06 2009-07-31   relativedelta(years=-9, months=-6, days=-25)
2 2000-01-07 2012-06-30  relativedelta(years=-12, months=-5, days=-23)

如果使用 apply 它应该更慢:

df['out'] = df.apply(lambda x:  date_diffs(x["d1"],x["d2"]) , axis=1)

推荐文章

user29747013 · 如何创建一个新的数据框架,其中包含原始数据框架中列的聚合列?

1 年前

Cam · Pandas列表日期到日期时间

1 年前

jjkennedy · Pandas文本文件导入:当每个文件中存在多个表时,自动选择1个表

1 年前

Sun Jar · 在另一个系列中查找当前df值的索引,并将其添加到列中

1 年前

dietzi96 · Pandas DataFrame.to_sql随机和静默地失败,没有错误消息

1 年前

Bijan · Pandas批量更新帐户字符串

1 年前

Kernel · TypeError:Index.reindex()收到意外的关键字参数fill_value'

1 年前

Kernel · 进入熊猫的定义。系列super().reindex

1 年前

adventurous_chip_55 · 如何引爆柱子

1 年前

RKIDEV · Panda迭代行并将第n行值乘以下一(n+1)行值

1 年前