如何從 Pandas DataFrame 提取列標題？

從 Pandas DataFrame 擷取欄位標題

Pandas DataFrame 是通用的資料結構，可實現高效率的資料操作與分析。一項常見任務涉及提取列標題，這對於獲取 DataFrame 結構的概述或進一步處理非常有用。

假設您有一個從使用者輸入匯入的 DataFrame，其中列的數量和名稱未知。要將列標題提取為列表，您可以利用以下方法：

方法1：使用DataFrame.columns.values

<code class="python">column_headers = list(my_dataframe.columns.values)</code>

my_dataframe.columns 傳回包含列標題的Index 物件。透過使用值將此索引轉換為列表，您將獲得表示列名稱的字串列表。

方法 2：使用 DataFrame.columns

<code class="python">column_headers = list(my_dataframe)</code>

This method 是 my_dataframe.columns.values 的簡寫符號，它直接將列轉換為列表。

範例用法

考慮DataFrame：

    y  gdp  cap
0   1    2    5
1   2    3    9
2   8    7    2
3   3    4    7
4   6    7    7
5   4    8    3
6   8    2    8
7   9    9   10
8   6    6    4
9  10   10    7

使用任一方法，您都會獲得以下列標題清單：

['y', 'gdp', 'cap']

使用任一方法，您都會獲得以下列標題清單：

以上是如何從 Pandas DataFrame 提取列標題？的詳細內容。更多資訊請關注PHP中文網其他相關文章！