首頁 > 後端開發 > Python教學 > 如何從 Pandas DataFrame 提取資料作為 Python 清單?

如何從 Pandas DataFrame 提取資料作為 Python 清單?

Barbara Streisand
發布: 2024-10-26 16:00:04
原創
727 人瀏覽過

How can I extract data from a Pandas DataFrame as a Python list?

從Pandas DataFrame 存取資料

要從Pandas DataFrame 檢索資料作為列表,有多種方法可用,取決於您是否想要存取特定的列或行。

以列表形式存取列

要以列表形式獲取列的內容,可以使用以下語法:

<code class="python">column_list = df['column_name'].tolist()</code>
登入後複製

這將傳回一個包含指定列的元素的Python 清單。

將行作為列表訪問

檢索一行作為列表,您可以使用iloc 或loc 方法,後跟.tolist():

<code class="python">row_list = df.iloc[row_index].tolist()  # row index-based
row_list = df.loc[row_label].tolist()  # row label-based</code>
登入後複製

範例

考慮DataFrame:

<code class="python">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']})</code>
登入後複製

以清單形式存取欄位:

<code class="python">cluster_list = df['cluster'].tolist()  # ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C']</code>
登入後複製

以列表形式存取行:

<code class="python">row_list = df.iloc[0].tolist()  # ['A', '1/1/2014', 1000, 4000, 'Y']</code>
登入後複製

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

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