首頁 >後端開發 >Python教學 >如何使用自訂函數從 Pandas DataFrame 中高效提取多列?

如何使用自訂函數從 Pandas DataFrame 中高效提取多列?

Patricia Arquette
Patricia Arquette原創
2024-11-03 00:55:29705瀏覽

How to Efficiently Extract Multiple Columns from a Pandas DataFrame using a Custom Function?

使用 Pandas 函數提取多列

本問題探討了使用自訂函數從 pandas DataFrame 中提取多列的問題。函數的傳回類型會出現問題,因為它需要與所需的輸出正確對齊。

最初,建議的方法是使用 df.iterrows() 迭代行。然而,後來發現這種方法明顯慢得多。因此,作者選擇將函數拆分為六個不同的 map(lambda ...) 呼叫來提取所需的列。

更有效的方法是利用 zip 函數來分配自訂函數的輸出同時到多個欄位。使用一個範例來說明此方法,其中將名為 powers 的函數應用於一列數字。此函數計算每個數字的六個冪值,並將結果指派給 DataFrame 中的六個新欄位。

這種方法既優雅又高效,並且避免了迭代 DataFrame 的行。這是基於自訂函數從 DataFrame 中提取多列的建議技術。

以上是如何使用自訂函數從 Pandas DataFrame 中高效提取多列?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn