如何编写DataFrame友好函数

Question

最近我一直在使用numpy的ndarray转换为pandas'DataFrame来满足我的数据分析需求。我注意到numpy函数似乎接受DataFrame个对象代替ndarray而没有问题。但是，当我尝试使用我在ndarray上编写的许多现有函数时，它们通常会在索引操作，广播等方面失败。我被迫通过ndarray传递基础df.values }。

是否有一些标准方法或一套指导方针可以使功能与DataFrame兼容？ numpy函数如何适应这两种类型？

Answer 1

我在numpy源代码中进行了挖掘，发现许多函数只是首先使用asarray或asanyarray等函数将输入转换为ndarray。

def numpyFunction(x, *args, **kwargs):
    x = np.asanyarray(x)
    ...