如何將函數應用於多個 Pandas DataFrame 欄位？

將函數套用至Pandas DataFrame 中的多個欄位

將函數依元素套用於Pandas DataFrame 中的多個欄位可能是一項常見任務。在此範例中，我們展示如何將函數應用於兩個特定列“col_1”和“col_2”，以建立新列“col_3”。

首先，定義要套用的函數。在本例中，我們有 get_sublist 它接受兩個參數：sta 和 end。此函數旨在根據輸入參數傳回預先定義清單 mylist 的子清單。

接下來，建立一個包含要使用的欄位的 Pandas DataFrame (df)。確保列具有適合您的函數的資料類型。

要將函數套用到多個列，請使用 Pandas 的 apply 函數。函數接受一個函數並將其套用於 DataFrame 的每一行。它需要指定軸參數，該參數應設為 1 以跨行應用該函數。

但是，使用具有位置參數的 apply 函數可能會導致錯誤。為了克服這個問題，我們可以使用匿名函數（也稱為 lambda 函數）作為要應用的參數。 lambda 函數可以使用名稱存取目前行的列值。

程式碼如下：

df['col_3'] = df.apply(lambda x: get_sublist(x.col_1, x.col_2), axis=1)

此程式碼建立一個新欄位“col_3”，其結果為將 get_sublist 函式套用到“col_1”和“col_2”的每一行。 lambda 函數確保將正確的參數傳遞給 get_sublist 函數。

因此，您將獲得具有所需輸出的 DataFrame，其中“col_3”中的每一行都包含一個基於對應值的子列表在“col_1”和“col_2”中。

以上是如何將函數應用於多個 Pandas DataFrame 欄位？的詳細內容。更多資訊請關注PHP中文網其他相關文章！