Pandas: วิธีเพิ่มคอลัมน์ตามเงื่อนไข

คุณสามารถใช้ไวยากรณ์ต่อไปนี้เพื่อรวมค่าของคอลัมน์ใน Pandas DataFrame ตามเงื่อนไข:

 df. loc [df[' col1 '] == some_value , ' col2 ']. sum ()

บทช่วยสอนนี้ให้ตัวอย่างหลายตัวอย่างการใช้งานจริงของไวยากรณ์นี้โดยใช้ Pandas DataFrame ต่อไปนี้:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'B', 'B', 'C'],
                   ' conference ': ['East', 'East', 'East', 'West', 'West', 'East'],
                   ' points ': [11, 8, 10, 6, 6, 5],
                   ' rebounds ': [7, 7, 6, 9, 12, 8]})

#view DataFrame
df

        team conference points rebounds
0 A East 11 7
1 A East 8 7
2 A East 10 6
3 B West 6 9
4 B West 6 12
5 C East 5 8

ตัวอย่างที่ 1: เพิ่มคอลัมน์ตามเงื่อนไข

รหัสต่อไปนี้แสดงวิธีการหาผลรวมของคะแนนสำหรับแถวที่ทีมเท่ากับ “A”:

 df. loc [df[' team '] == ' A ', ' points ']. sum ()

29

ตัวอย่างที่ 2: เพิ่มคอลัมน์ตามเงื่อนไขหลายข้อ

รหัสต่อไปนี้แสดงวิธีการหาผลรวมของคะแนนสำหรับเส้นที่ทีมเท่ากับ “A” และ การประชุมเท่ากับ “ตะวันออก”:

 df. loc [(df[' team '] == ' A ') & (df[' conference '] == ' East '), ' points ']. sum ()

29

ตัวอย่างที่ 3: เพิ่มคอลัมน์ตามเงื่อนไขข้อใดข้อหนึ่ง

รหัสต่อไปนี้แสดงวิธีการหาผลรวมของคะแนนสำหรับเส้นที่ทีมเท่ากับ “A” หรือ “B”:

 df. loc [df[' team ']. isin ([' A ',' B ']), ' dots ']. sum ()

41

คุณสามารถดูบทช่วยสอนเกี่ยวกับแพนด้าเพิ่มเติมได้ใน หน้านี้