Отримайте загальний стовпець Pandas


107

Ціль

У мене є кадр даних панди, як показано нижче, з кількома стовпцями і хотіли б отримати в загальній складності колони, MyColumn.


Кадр даних -df:

print df

           X           MyColumn  Y              Z   
0          A           84        13.0           69.0   
1          B           76         77.0          127.0   
2          C           28         69.0           16.0   
3          D           28         28.0           31.0   
4          E           19         20.0           85.0   
5          F           84        193.0           70.0   

Моя спроба :

Я спробував отримати суму стовпця за допомогою groupbyта .sum():

Total = df.groupby['MyColumn'].sum()

print Total

Це викликає таку помилку:

TypeError: 'instancemethod' object has no attribute '__getitem__'

Очікуваний вихід

Я очікував, що результат буде наступним:

319

Або я хотів би dfвідредагувати нову rowназву, TOTALщо містить загальну кількість:

           X           MyColumn  Y              Z   
0          A           84        13.0           69.0   
1          B           76         77.0          127.0   
2          C           28         69.0           16.0   
3          D           28         28.0           31.0   
4          E           19         20.0           85.0   
5          F           84        193.0           70.0   
TOTAL                  319

11
Для ілюстрації того, чому панди не є пітонічними, дивіться не далі, як плутанина щодо того, як просто підсумовувати стовпчик.
користувач1416227

Відповіді:


214

Ви повинні використовувати sum:

Total = df['MyColumn'].sum()
print (Total)
319

Тоді ви використовуєте locз Series, в цьому випадку індекс повинен бути встановлений таким же, як і конкретний стовпець, який потрібно підсумувати:

df.loc['Total'] = pd.Series(df['MyColumn'].sum(), index = ['MyColumn'])
print (df)
         X  MyColumn      Y      Z
0        A      84.0   13.0   69.0
1        B      76.0   77.0  127.0
2        C      28.0   69.0   16.0
3        D      28.0   28.0   31.0
4        E      19.0   20.0   85.0
5        F      84.0  193.0   70.0
Total  NaN     319.0    NaN    NaN

тому що якщо ви передасте скаляр, значення всіх рядків будуть заповнені:

df.loc['Total'] = df['MyColumn'].sum()
print (df)
         X  MyColumn      Y      Z
0        A        84   13.0   69.0
1        B        76   77.0  127.0
2        C        28   69.0   16.0
3        D        28   28.0   31.0
4        E        19   20.0   85.0
5        F        84  193.0   70.0
Total  319       319  319.0  319.0

Ще два рішення є at, і ixперегляньте додатки нижче:

df.at['Total', 'MyColumn'] = df['MyColumn'].sum()
print (df)
         X  MyColumn      Y      Z
0        A      84.0   13.0   69.0
1        B      76.0   77.0  127.0
2        C      28.0   69.0   16.0
3        D      28.0   28.0   31.0
4        E      19.0   20.0   85.0
5        F      84.0  193.0   70.0
Total  NaN     319.0    NaN    NaN

df.ix['Total', 'MyColumn'] = df['MyColumn'].sum()
print (df)
         X  MyColumn      Y      Z
0        A      84.0   13.0   69.0
1        B      76.0   77.0  127.0
2        C      28.0   69.0   16.0
3        D      28.0   28.0   31.0
4        E      19.0   20.0   85.0
5        F      84.0  193.0   70.0
Total  NaN     319.0    NaN    NaN

Примітка. Починаючи з Pandas v0.20, ixвін застарів. Використовуйте locабо ilocзамість цього.


Це чудово :) Дякую за пояснення, чи можу я запитати, що .locробити у наведеному вище прикладі?
LearningToJava


atпрацює і для налаштування з розширенням, див. останнє редагування.
jezrael

Спасибі, чи є якийсь кращий метод?
LearningToJava

1
Гммм, документи говорять The .loc/.ix/[] operations can perform enlargement when setting a non-existant key for that axis., так locчи ixабо []. у наступному розділі пише at may enlarge the object in-place as above if the indexer is missing.Так що всі методи хороші, але atя думаю , що це найшвидше.
jezrael

22

Ще один варіант, з яким ви можете скористатися тут:

df.loc["Total", "MyColumn"] = df.MyColumn.sum()

#         X  MyColumn      Y       Z
#0        A     84.0    13.0    69.0
#1        B     76.0    77.0   127.0
#2        C     28.0    69.0    16.0
#3        D     28.0    28.0    31.0
#4        E     19.0    20.0    85.0
#5        F     84.0   193.0    70.0
#Total  NaN    319.0     NaN     NaN

Ви також можете використовувати append()метод:

df.append(pd.DataFrame(df.MyColumn.sum(), index = ["Total"], columns=["MyColumn"]))

введіть тут опис зображення


Оновлення:

Якщо вам потрібно додати суму для всіх числових стовпців, ви можете виконати одне з наступних дій:

Використовуйте appendдля цього функціонально (не змінює початковий кадр даних):

# select numeric columns and calculate the sums
sums = df.select_dtypes(pd.np.number).sum().rename('total')

# append sums to the data frame
df.append(sums)
#         X  MyColumn      Y      Z
#0        A      84.0   13.0   69.0
#1        B      76.0   77.0  127.0
#2        C      28.0   69.0   16.0
#3        D      28.0   28.0   31.0
#4        E      19.0   20.0   85.0
#5        F      84.0  193.0   70.0
#total  NaN     319.0  400.0  398.0

Використовуйте locдля вимкнення кадру даних на місці:

df.loc['total'] = df.select_dtypes(pd.np.number).sum()
df
#         X  MyColumn      Y      Z
#0        A      84.0   13.0   69.0
#1        B      76.0   77.0  127.0
#2        C      28.0   69.0   16.0
#3        D      28.0   28.0   31.0
#4        E      19.0   20.0   85.0
#5        F      84.0  193.0   70.0
#total  NaN     638.0  800.0  796.0

Як щодо суми всіх стовпців?
FaCoffee

9

Подібно до отримання довжини фрейму даних, len(df)для панд та пластів працювали наступні:

Total = sum(df['MyColumn'])

або альтернативно

Total = sum(df.MyColumn)
print Total

2

Є два способи підсумовувати стовпчик

набор даних = pd.read_csv ("data.csv")

1: сума (data.Column_name)

2: набір даних ['Ім'я стовпця']. Sum ()

Якщо в цьому є якась проблема, будь ласка, виправте мене.


1

Як інший варіант, ви можете зробити щось на кшталт нижче

Group   Valuation   amount
    0   BKB Tube    156
    1   BKB Tube    143
    2   BKB Tube    67
    3   BAC Tube    176
    4   BAC Tube    39
    5   JDK Tube    75
    6   JDK Tube    35
    7   JDK Tube    155
    8   ETH Tube    38
    9   ETH Tube    56

Нижче сценарію ви можете використовувати для наведених вище даних

import pandas as pd    
data = pd.read_csv("daata1.csv")
bytreatment = data.groupby('Group')
bytreatment['amount'].sum()
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.