從 Pandas DataFrame 擷取列標題

從 Pandas DataFrame 擷取列標題

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

column_headers = list(my_dataframe.columns.values)

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

column_headers = list(my_dataframe)

此方法為my_dataframe.columns.values.符號，它直接將列轉換為列表。

用法範例

    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']

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