如何编写DataFrame友好功能

Question

最近，我一直在使用numpy的ndarray转换熊猫的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) 
    ...