Rumah >pembangunan bahagian belakang >Tutorial Python >Bagaimanakah cara saya mengira jumlah baris lajur tertentu dalam Pandas DataFrame?

Bagaimanakah cara saya mengira jumlah baris lajur tertentu dalam Pandas DataFrame?

Susan Sarandon
Susan Sarandonasal
2024-11-10 12:51:02607semak imbas

How do I calculate the row sum of specific columns in a Pandas DataFrame?

Penjumlahan Baris Lajur Diberi dalam Pandas DataFrame

Dalam perpustakaan Pandas Python, kita sering menghadapi keperluan untuk mengira jumlah lajur tertentu dalam sebuah DataFrame. Untuk mencapai matlamat ini dengan berkesan, kita mesti mempertimbangkan parameter dan operasi yang sesuai.

Mari kita pertimbangkan DataFrame berikut:

df = pd.DataFrame({'a': [1, 2, 3],
                   'b': [2, 3, 4],
                   'c': ['dd', 'ee', 'ff'],
                   'd': [5, 9, 1]})

Objektif kami adalah untuk menambah lajur 'e' yang mewakili jumlah lajur 'a', 'b' dan 'd'. Walaupun secara intuitif, seseorang mungkin mendekati ini dengan sesuatu seperti:

df['e'] = df[['a', 'b', 'd']].map(sum)

kaedah ini gagal menghasilkan output yang diingini.

Pendekatan yang betul melibatkan penggunaan fungsi sum() dengan parameter berikut:

  • axis=1: Menentukan penjumlahan harus dilakukan di sepanjang baris (mendatar).
  • numeric_only=True: Memastikan bahawa hanya lajur angka dipertimbangkan dalam operasi, tidak termasuk lajur bukan angka seperti 'c'.

Menggunakan pendekatan ini menghasilkan perkara berikut hasil:

df['e'] = df.sum(axis=1, numeric_only=True)

Output:

   a  b   c  d   e
0  1  2  dd  5   8
1  2  3  ee  9  14
2  3  4  ff  1   8

Sebagai alternatif, jika kita ingin mengira jumlah lajur tertentu sahaja, kita boleh membuat senarai lajur tersebut dan menghapuskan lajur yang kita buat tidak perlu menggunakan remove() kaedah.

col_list = list(df)
col_list.remove('d')

df['e'] = df[col_list].sum(axis=1)

Output:

   a  b   c  d  e
0  1  2  dd  5  3
1  2  3  ee  9  5
2  3  4  ff  1  7

Dengan menggunakan operasi ini, kami boleh menjumlahkan baris dengan berkesan untuk lajur tertentu dalam Pandas DataFrame, memastikan analisis data yang tepat dan cekap.

Atas ialah kandungan terperinci Bagaimanakah cara saya mengira jumlah baris lajur tertentu dalam Pandas DataFrame?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

Kenyataan:
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn