ฉันมี DataFrame ต่อไปนี้:
In [1]:
import pandas as pd
df = pd.DataFrame({'a': [1,2,3], 'b': [2,3,4], 'c':['dd','ee','ff'], 'd':[5,9,1]})
df
Out [1]:
a b c d
0 1 2 dd 5
1 2 3 ee 9
2 3 4 ff 1
ผมอยากจะเพิ่มคอลัมน์'e'ซึ่งเป็นผลรวมของคอลัมน์'a', และ'b''d'
จะข้ามฟอรัมฉันคิดว่าสิ่งนี้จะทำงาน:
df['e'] = df[['a','b','d']].map(sum)แต่มันก็ไม่ได้
ฉันต้องการทราบการดำเนินการที่เหมาะสมกับรายการของคอลัมน์['a','b','d']และdfเป็นอินพุต
คำตอบ:
คุณสามารถsumและตั้งค่าพารามิเตอร์axis=1เพื่อหาผลรวมของแถวซึ่งจะไม่สนใจคอลัมน์ตัวเลข:
In [91]:
df = pd.DataFrame({'a': [1,2,3], 'b': [2,3,4], 'c':['dd','ee','ff'], 'd':[5,9,1]})
df['e'] = df.sum(axis=1)
df
Out[91]:
a b c d e
0 1 2 dd 5 8
1 2 3 ee 9 14
2 3 4 ff 1 8หากคุณต้องการรวมคอลัมน์เฉพาะคุณสามารถสร้างรายการคอลัมน์และลบคอลัมน์ที่คุณไม่สนใจ:
In [98]:
col_list= list(df)
col_list.remove('d')
col_list
Out[98]:
['a', 'b', 'c']
In [99]:
df['e'] = df[col_list].sum(axis=1)
df
Out[99]:
a b c d e
0 1 2 dd 5 3
1 2 3 ee 9 5
2 3 4 ff 1 7หากคุณมีผลรวมเพียงไม่กี่คอลัมน์คุณสามารถเขียน:
df['e'] = df['a'] + df['b'] + df['d']สิ่งนี้จะสร้างคอลัมน์ใหม่eพร้อมค่า:
a b c d e
0 1 2 dd 5 8
1 2 3 ee 9 14
2 3 4 ff 1 8สำหรับรายการคอลัมน์ที่ยาวกว่าต้องการคำตอบของ EdChum
สร้างรายการชื่อคอลัมน์ที่คุณต้องการเพิ่ม
df['total']=df.loc[:,list_name].sum(axis=1)หากคุณต้องการผลรวมสำหรับแถวที่แน่นอนให้ระบุแถวโดยใช้ ':'
นี่เป็นวิธีที่ง่ายกว่าโดยใช้ iloc เพื่อเลือกคอลัมน์ที่จะรวม:
df['f']=df.iloc[:,0:2].sum(axis=1)
df['g']=df.iloc[:,[0,1]].sum(axis=1)
df['h']=df.iloc[:,[0,3]].sum(axis=1)ผลิต:
a b c d e f g h
0 1 2 dd 5 8 3 3 6
1 2 3 ee 9 14 5 5 11
2 3 4 ff 1 8 7 7 4ฉันไม่สามารถหาวิธีรวมช่วงและคอลัมน์เฉพาะที่ใช้งานได้เช่น:
df['i']=df.iloc[:,[[0:2],3]].sum(axis=1)
df['i']=df.iloc[:,[0:2,3]].sum(axis=1)ไวยากรณ์ต่อไปนี้ช่วยฉันเมื่อฉันมีคอลัมน์ตามลำดับ
awards_frame.values[:,1:4].sum(axis =1)คุณสามารถส่งชื่อไฟล์ของคุณไปยังฟังก์ชั่นต่อไปนี้:
def sum_frame_by_column(frame, new_col_name, list_of_cols_to_sum):
frame[new_col_name] = frame[list_of_cols_to_sum].astype(float).sum(axis=1)
return(frame)ตัวอย่าง :
ฉันมีชื่อไฟล์ (awards_frame) ดังต่อไปนี้:
... และฉันต้องการสร้างคอลัมน์ใหม่ที่แสดงผลรวมของรางวัลสำหรับแต่ละแถว :
การใช้งาน :
ฉันแค่ส่งawards_frameของฉันเข้าสู่ฟังก์ชันรวมทั้งระบุชื่อคอลัมน์ใหม่และรายการชื่อคอลัมน์ที่จะสรุป:
sum_frame_by_column(awards_frame, 'award_sum', ['award_1','award_2','award_3'])ผลลัพธ์ :
วิธีที่สั้นและง่ายที่สุดที่นี่คือการใช้
df.eval('e = a + b + d')