Какая аннотация для функции aggregation_function в pd.Series.agg (aggregation_function)? - PullRequest
1 голос
/ 09 июля 2020

Сильно упрощено, у меня есть такой код:

import pandas as pd

def aggregate_series(series: pd.Series, agg_function: Callable) -> Any:
    return series.agg(agg_function)

Я хотел бы лучше аннотировать это. Какие параметры agg_function? Тип возврата не может быть указан лучше, чем Any, верно?

Примеры для agg_functions:

  • sum, min, max (Python встроенные)
  • np.mean, np.median (numpy функции)
  • Все, что угодно: lambda n: max(el for el in n if el < 2)

Обратите внимание, что max может также работать с серией строк. А пользовательская функция, конечно же, может работать с любым типом объекта.

Правильно ли agg_function: Callable[Sequence[Any], Any]? Лучшее, что я могу сделать?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...