rapidsai/cudf

[FEA] efficient way to truncate by month

Open

#18,650 opened on 2025年5月3日

GitHub で見る
 (5 comments) (0 reactions) (1 assignee)C++ (6,000 stars) (735 forks)batch import
Pythonfeature requestgood first issuelibcudf

説明

Is there an efficient way to truncate a Series by month / quarter in cuDF?

https://docs.rapids.ai/api/cudf/latest/user_guide/api_docs/api/cudf.core.series.datetimeproperties.floor/ dt.floor doesn't support month / year / quarter

In pandas I can either, depending on the dtype backend:

  • drop to numpy and use astype('datetime64["3M"]'))
  • drop to PyArrow and use pc.floor_temporal

Any suggestion for cuDF?

Example: given

import cudf
from datetime import datetime

s = cudf.Series([datetime(2020, 2, 4)])
s.dt.truncate('3 months')  # Not valid syntax, I know :)

I'd like to end up with

'2020-01-01'

コントリビューターガイド

技術スタック
pythoncpp
領域
dataperformance
Issue 種別
feature
難度新しい貢献者にとっての実装難度です。1 は非常に小さな変更、5 は専門的な作業です。
3
推定時間経験ある貢献者が調査、実装、テスト、pull request 準備にかけるおおよその時間範囲です。
half day
活動状況Issue が今どれだけ取り組みやすいかを示します。新しい、活発、古い、ブロック中、またはメンテナー入力待ちなどです。
blocked
明確さ期待される変更、受け入れ条件、次の手順がどれだけ明確かを示します。
clear
前提条件
Basic PythonBasic cuDF
初心者向け度初回貢献者にどれだけ取り組みやすいかを 1-100 で推定したスコアです。
40
調査方針
Investigate the current implementation of truncation in cuDF's datetime functionality. Look at the `dt.floor` method and how it currently supports only certain frequencies. The issue suggests a need for month/quarter truncation. Review the relevant C++ source files in `cpp/src/datetime/` and Python wrappers in `python/cudf/cudf/core/series/datetime.py`. Consider if a new kernel is needed or if existing kernels can be extended. Also check if any assignee has started work or if there are discussions in the comments that clarify the approach. The user mentions pandas and PyArrow alternatives, so compare those implementations.