首頁  >  文章  >  後端開發  >  如何從 Pandas DataFrames 中提取資料作為清單?

如何從 Pandas DataFrames 中提取資料作為清單?

Mary-Kate Olsen
Mary-Kate Olsen原創
2024-10-26 21:26:29448瀏覽

How to Extract Data from Pandas DataFrames as Lists?

從 Pandas DataFrame 取得清單

使用 Pandas DataFrame 時,通常需要將資料提取為清單以進行進一步分析或處理。以下是您問題的答案:

以列表形式檢索列

要以列表形式檢索列,只需將其作為Pandas 系列訪問,然後進行轉換使用.tolist() 將其加入到列表中。

import pandas as pd

df = pd.DataFrame({
    'cluster': ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
    'load_date': ['1/1/2014', '2/1/2014', '3/1/2014', '4/1/2014', '4/1/2014', '4/1/2014', '7/1/2014', '8/1/2014', '9/1/2014'],
    'budget': [1000, 12000, 36000, 15000, 12000, 90000, 22000, 30000, 53000],
    'actual': [4000, 10000, 2000, 10000, 11500, 11000, 18000, 28960, 51200],
    'fixed_price': ['Y', 'Y', 'Y', 'N', 'N', 'N', 'N', 'N', 'N']
})

df['cluster'].tolist()

這將傳回一個包含「簇」列中的值的清單:

['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C']

檢索行或列作為清單

要以清單形式擷取行或列的全部內容,請使用.values 屬性。

例如,要以列表形式檢索第 2 行:

df.iloc[1].values.tolist()

並將第 3 列作為列表檢索:

df['actual'].values.tolist()

以上是如何從 Pandas DataFrames 中提取資料作為清單?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

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