Bagaimana untuk Menghimpunkan Baris DataFrame dan Mencipta Senarai Nilai Menggunakan Panda?

Cara Mengubah Baris Dataframe menjadi Senarai Berkumpulan Menggunakan Pandas Groupby

Masalah:

Diberikan DataFrame dengan berbilang lajur, matlamatnya adalah untuk mengumpulkan baris berdasarkan lajur tertentu (cth., lajur pertama) dan kumpulkan nilai daripada lajur lain (cth. lajur kedua) ke dalam senarai dalam setiap kumpulan. Contohnya:

Frame Data Asal:

a	b
A	1
A	2
B	5
B	5
B	4
C	6

Hasil yang Diingini:

a	new
A	[1, 2]
B	[5, 5, 4]
C	[6]

Penyelesaian:

Untuk mencapai matlamat ini hasil, fungsi kumpulan mengikut panda boleh digunakan bersama-sama dengan kaedah apply() untuk menukar setiap kumpulan menjadi a senarai.

import pandas as pd

# Create a DataFrame
df = pd.DataFrame({'a': ['A', 'A', 'B', 'B', 'B', 'C'], 'b': [1, 2, 5, 5, 4, 6]})

# Group by the "a" column and create lists of "b" values
grouped_list = df.groupby('a')['b'].apply(list)

# Reset the index to obtain a DataFrame
df1 = grouped_list.reset_index(name='new')

Penjelasan:

• groupby('a'): Himpunkan DataFrame mengikut lajur "a" .
• apply(list): Gunakan fungsi list() untuk setiap kumpulan, menukar nilai "b" kepada senarai.
• reset_index(name='new'): Menetapkan semula indeks dan memberikan nama lajur baharu "baharu" kepada senarai berkumpulan.

DataFrame yang terhasil, df1, akan mengandungi kumpulan yang dikehendaki senarai:

   a        new
0  A     [1, 2]
1  B  [5, 5, 4]
2  C        [6]

Atas ialah kandungan terperinci Bagaimana untuk Menghimpunkan Baris DataFrame dan Mencipta Senarai Nilai Menggunakan Panda?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!