如何編寫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) 
    ...