首頁 >後端開發 >Python教學 >如何使用 Pandas 解壓縮 DataFrame 中的嵌套清單?

如何使用 Pandas 解壓縮 DataFrame 中的嵌套清單?

Barbara Streisand
Barbara Streisand原創
2024-11-13 07:28:02259瀏覽

How to Unpack Nested Lists in DataFrames with Pandas?

使用Pandas 分解DataFrame 中的巢狀清單

挑戰:將包含清單的DataFrame 單元格轉換為每個值的單獨行。

考慮一個 DataFrame,其中包含一列對手名稱和對應的最近鄰居列,以列表形式表示。目標是解壓縮這些列表,將每個鄰居堆疊為對應對手的單獨行。

解:

Pandas 使用explode() 簡化了此操作0.25版本中的方法:

import pandas as pd

# Sample DataFrame
df = (pd.DataFrame({'name': ['A.J. Price'] * 3, 
                    'opponent': ['76ers', 'blazers', 'bobcats'], 
                    'nearest_neighbors': [['Zach LaVine', 'Jeremy Lin', 'Nate Robinson', 'Isaia']] * 3})
      .set_index(['name', 'opponent']))

# Explode the 'nearest_neighbors' column
df = df.explode('nearest_neighbors')

# Print the exploded DataFrame
print(df)

輸出:

                    nearest_neighbors
name       opponent                  
A.J. Price 76ers          Zach LaVine
           76ers           Jeremy Lin
           76ers        Nate Robinson
           76ers                Isaia
           blazers        Zach LaVine
           blazers         Jeremy Lin
           blazers      Nate Robinson
           blazers              Isaia
           bobcats        Zach LaVine
           bobcats         Jeremy Lin
           bobcats      Nate Robinson
           bobcats              Isaia

以上是如何使用 Pandas 解壓縮 DataFrame 中的嵌套清單?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

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