Bagaimana untuk Membentuk Semula Data Panda daripada Format Panjang ke Lebar dengan Berbilang Pembolehubah Menggunakan Fungsi Pivot?

Pembentukan Semula Panda Panjang ke Lebar dengan Berbilang Pembolehubah

Menukar data daripada format panjang ke lebar dalam Panda boleh menjadi mencabar, terutamanya apabila berbilang pembolehubah terlibat. Soalan ini meneroka kaedah untuk membentuk semula data menggunakan fungsi pangsi.

Data asal yang disediakan ialah:

  Salesman  Height   product      price
  Knut      6        bat          5
  Knut      6        ball         1
  Knut      6        wand         3
  Steve     5        pen          2

Format lebar yang dikehendaki ialah:

Salesman  Height    product_1  price_1  product_2 price_2 product_3 price_3  
  Knut      6        bat          5       ball      1        wand      3
  Steve     5        pen          2        NA       NA        NA       NA

Satu pendekatan, seperti yang dicadangkan oleh Chris Albon, melibatkan penggunaan fungsi pangsi seperti berikut:

df.pivot(index='Salesman', columns='product', values='price')

Pendekatan ini mencipta indeks berbilang peringkat, dengan lajur Jurujual dan produk sebagai indeks baris dan lajur, masing-masing. Lajur harga menjadi nilai.

Kerangka data yang terhasil ialah:

product      bat  ball  wand
Salesman                 
Knut          5    1     3
Steve         2   NaN   NaN

Untuk mendapatkan format yang dikehendaki, langkah tambahan diperlukan untuk menyusun lajur dan mengekstrak nilai produk dan harga ke dalam lajur berasingan. Ini boleh dicapai menggunakan fungsi tindanan dan reset_index seperti berikut:

df.pivot(index='Salesman', columns='product', values='price') \
   .stack().reset_index() \
   .rename(columns={'level_1':'product', 0:'price'})

Hasil akhir ialah format lebar yang diingini.

Atas ialah kandungan terperinci Bagaimana untuk Membentuk Semula Data Panda daripada Format Panjang ke Lebar dengan Berbilang Pembolehubah Menggunakan Fungsi Pivot?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!