首頁 > 後端開發 > Python教學 > 如何將 Pandas DataFrame 列或行轉換為清單?

如何將 Pandas DataFrame 列或行轉換為清單?

Mary-Kate Olsen
發布: 2024-10-27 01:02:30
原創
997 人瀏覽過

How do you convert a Pandas DataFrame column or row to a list?

如何將 Pandas DataFrame 列或行轉換為列表

在 pandas DataFrame 中,每一列都由一個 pandas Series 物件表示。要取得列的列表表示形式,可以在 Series 物件上使用 tolist() 方法。例如:

<code class="python">import pandas as pd

data_dict = {'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 = pd.DataFrame(data_dict)
cluster_list = df['cluster'].tolist()
print(cluster_list)</code>
登入後複製

輸出:

['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C']
登入後複製

也可以直接使用list()將Series物件轉換為清單:

<code class="python">cluster_list = list(df['cluster'])</code>
登入後複製

來取得整行的清單表示,您可以使用DataFrame 的iloc() 方法存取它。

<code class="python">row1_list = df.iloc[0].tolist()
print(row1_list)</code>
登入後複製

輸出:

[1000, '4000', 'Y']
登入後複製

同樣,您可以轉換整行直接到列表:

<code class="python">row1_list = list(df.iloc[0])</code>
登入後複製

以上是如何將 Pandas DataFrame 列或行轉換為清單?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板