Сильно упрощено, у меня есть такой код:
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]
? Лучшее, что я могу сделать?