Python: как получить сумму значений на основе разных столбцов

Question

У меня есть рамка данных df, например:

df  name     city
0   John    New York
1   Carl    New York
2   Carl     Paris
3   Eva      Paris
4   Eva      Paris
5   Carl     Paris

Я хочу узнать общее количество людей в разных городах

df2  city      number
0   New York     2
1    Paris       3

или количество людейс одноименным названием в городах

df2    name     city      number
0      John    New York     1
1      Eva      Paris       2
2      Carl     Paris       2
3      Eva      New York    0

Answer 1

Для подсчета количества людей с разными именами в одном городе:

groups = df.groupby('city').count().reset_index()

Для подсчета количества людей с одинаковыми именами в разных городах:

groups = df.groupby('city').count().reset_index()

Answer 2

Я считаю, что нужно GroupBy.size:

df1 = df.groupby(['city']).size().reset_index(name='number')
print (df1)
       city  number
0  New York       2
1     Paris       4

---

df2 = df.groupby(['name','city']).size().reset_index(name='number')
print (df2)
   name      city  number
0  Carl  New York       1
1  Carl     Paris       2
2   Eva     Paris       2
3  John  New York       1

При необходимости для всех комбинаций добавляется одно решение unstack и stack:

df3=df.groupby(['name','city']).size().unstack(fill_value=0).stack().reset_index(name='count')
print (df3)
   name      city  number
0  Carl  New York       1
1  Carl     Paris       2
2   Eva  New York       0
3   Eva     Paris       2
4  John  New York       1
5  John     Paris       0

Или reindex с MultiIndex.from_product:

df2 = df.groupby(['name','city']).size()
mux = pd.MultiIndex.from_product(df2.index.levels, names=df2.index.names)
df2 = df2.reindex(mux, fill_value=0).reset_index(name='number')
print (df2)
   name      city  number
0  Carl  New York       1
1  Carl     Paris       2
2   Eva  New York       0
3   Eva     Paris       2
4  John  New York       1
5  John     Paris       0