Создайте несколько новых фреймов данных или словарей из одного фрейма данных

Question

У меня есть такой фрейм данных:

evt    pcle    bin_0    bin_1    bin_2    ...    bin_49
 1      pi      1        0         0               0 
 1      pi      1        0         0               0 
 1      k       0        0         0               1 
 1      pi      0        0         1               0 
 2      pi      0        0         1               0 
 2      k       0        1         0               0 
 3      J       0        1         0               0 
 3      pi      0        0         0               1 
 3      pi      1        0         0               0 
 3      k       0        1         0               0 
 ...
 5000   J       0        0         1               0 
 5000   pi      0        1         0               0 
 5000   k       0        0         0               1

С этой информацией я хочу создать несколько других фреймов данных df_ {evt} (или, может быть, словари должны быть лучше?):

df_1 : 
pcle    cant    bin_0    bin_1    bin_2   ...    bin_49        
 pi      3        2        0        1              0
  k      1        0        0        0              1

df_2 : 
pcle    cant    bin_0    bin_1    bin_2   ...    bin_49        
 pi      1        0        0        1              0
  k      0        1        0        0              0

Всего было бы 5000 информационных фреймов (по 1 на каждый evt), где в каждом из них:

*the column "cant" has the ocurrences of "pcle" in the particular "evt". 

*bin_0 ... bin_49 have the sum of the values for this particular "pcle" in 
 the particular "evt".

Каков наилучший способ достижения этой цели?

Answer 1

Вот возможное решение:

import pandas as pd
import numpy as np
columns = ["evt", "pcle", "bin_0", "bin_1", "bin_2", "bin_3"]
data = [[1, "pi", 1, 0, 0, 0],
        [1, "pi", 0, 0, 0, 0],
        [1, "k", 0, 0, 0, 1],
        [1, "pi", 0, 0, 1, 0],
        [2, "pi", 0, 0, 1, 0],
        [2, "k", 0, 1, 0, 0],
        [3, "J", 0, 1, 0, 0],
        [3, "pi", 0, 0, 0, 1],
        [3, "pi", 1, 0, 0, 0],
        [3, "k", 0, 1, 0, 0]]

df = pd.DataFrame(data=data, columns=columns)

# group your data by the columns you want
grouped = df.groupby(["evt", "pcle"])

# compute the aggregates for the bin_X
df_t = grouped.aggregate(np.sum)

# move pcle from index to column
df_t.reset_index(level=["pcle"], inplace=True)

# count occurrences of pcle
df_t["cant"] = grouped.size().values

# filter evt with .loc
df_t.loc[1]

Если вы хотите превратить его в словарь, вы можете выполнить:

d = {i:j.reset_index(drop=True) for i, j in df_t.groupby(df_t.index)}