python - 为“cudf”做“diff”的最有效方法是什么

Question

该rapids.ai cudf类型与 有点兼容pandas，但这里有一个奇怪的不兼容。cudf.Series有一个.diff()方法，但 acudf.DataFrame似乎没有。这是非常烦人的（例如，考虑一个股票价格数据框，其中的列对应于工具）。当然，有一些笨拙的方法可以解决这个问题（转换为 pandas 数据框并回到脑海中），但我想知道规范的方法是什么。有什么建议吗？

Answer 1

cuDF Python 涵盖了 pandas API 的大部分内容，但存在一些差距（正如您在此处遇到的那样）。

今天，diff在每一列上运行并返回数据框的最简单方法如下：

cudf.DataFrame({col: df[col].diff() for col in df.columns})