Groupby в Python Pandas: быстрый способ ⇐ Python

[Anonymous]

Я хочу улучшить время группировки в Python Pandas.
У меня есть этот код:

Код: Выделить всё

df["Nbcontrats"] = df.groupby(['Client', 'Month'])['Contrat'].transform(len)

Цель — подсчитать, сколько контрактов заключил клиент за месяц, и добавить эту информацию в новый столбец (

Код: Выделить всё

Nbcontrats

).

• Код: Выделить всё

  Client

  : код клиента

• Код: Выделить всё

  Month

  : месяц извлечения данных

• Код: Выделить всё

  Contrat

  : номер контракта

Я хочу улучшить время. Ниже я работаю только с частью моих реальных данных:

Код: Выделить всё

%timeit df["Nbcontrats"] = df.groupby(['Client', 'Month'])['Contrat'].transform(len)
1 loops, best of 3: 391 ms per loop

df.shape
Out[309]: (7464, 61)

Как сократить время выполнения?

Подробнее здесь: https://stackoverflow.com/questions/381 ... s-fast-way