如何使用使用者輸入從 Pandas DataFrame 中提取列標題？

從 Pandas DataFrame 提取列標題

從 Pandas DataFrame 中取得列標題清單是資料分析的常見操作。在本文中，我們將示範如何在根據使用者輸入產生 DataFrame 時實現此目的，確保與未知數量或名稱的列相容。

DataFrame 列標題提取

要從DataFrame 中獲取列標題列表，可以使用以下方法：

• columns. values：此屬性傳回一個列標籤數組，它可以使用list(my_dataframe.columns.values) 轉換為列表。
• 直接轉換：或者，您可以簡單地將 DataFrame 轉換為 list(my_dataframe) 列表，這將導致列標題列表，後跟 DataFrame 值。

範例

考慮以下DataFrame：

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

data = {
    'y': [1, 2, 8, 3, 6, 4, 8, 9, 6, 10],
    'gdp': [2, 3, 7, 4, 7, 8, 2, 9, 6, 10],
    'cap': [5, 9, 2, 7, 7, 3, 8, 10, 4, 7]
}

df = pd.DataFrame(data)</code>

取得列標題

取得列標題headers = list(df.columns.values) print(headers) # Output: ['y', 'gdp', 'cap']<p></p>使用直接轉換：<pre class="brush:php;toolbar:false">&lt;code class=&quot;python&quot;&gt;headers = list(df) print(headers) # Output: ['y', 'gdp', 'cap']&lt;/code&gt;</pre> <p></p>兩種方法都會提供列標題清單: ['y', 'gdp', '上限'] .

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