Bagaimana untuk Mengekstrak Pengepala Lajur daripada Pandas DataFrame dengan Input Pengguna?

Mengekstrak Pengepala Lajur daripada Pandas DataFrame

Mendapatkan senarai pengepala lajur daripada Pandas DataFrame ialah operasi biasa untuk analisis data. Dalam artikel ini, kami akan menunjukkan cara untuk mencapai ini apabila DataFrame dijana daripada input pengguna, memastikan keserasian dengan nombor atau nama lajur yang tidak diketahui.

Pengeluaran Pengepala Lajur DataFrame

Untuk memperoleh senarai pengepala lajur daripada DataFrame, anda boleh menggunakan yang berikut:

• columns.values: Atribut ini mengembalikan tatasusunan label lajur, yang boleh ditukar kepada senarai menggunakan list(my_dataframe.columns.values).
• Directly Casting: Sebagai alternatif, anda boleh menghantar DataFrame ke senarai dengan list(my_dataframe) Ini akan mengakibatkan senarai pengepala lajur diikuti dengan nilai DataFrame.

Contoh

Pertimbangkan DataFrame berikut:

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

Mendapatkan Pengepala Lajur

Menggunakan kaedah columns.values:

<code class="python">headers = list(df.columns.values)
print(headers)

# Output: ['y', 'gdp', 'cap']</code>

Menggunakan penghantaran langsung:

<code class="python">headers = list(df)
print(headers)

# Output: ['y', 'gdp', 'cap']</code>

Kedua-dua pendekatan akan menyediakan senarai pengepala lajur : ['y', 'gdp', 'cap'].

Atas ialah kandungan terperinci Bagaimana untuk Mengekstrak Pengepala Lajur daripada Pandas DataFrame dengan Input Pengguna?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!