Определите тип данных при создании пользовательской функции ⇐ Python

[Anonymous]

У меня есть пользовательская функция ниже

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

import pandas as pd
def MyFn(DF : pd.DataFrame) -> float :
return DF['Col_A'].values[1] - DF['Col_B'].values[1]

Однако я хочу заставить пользователя предоставить фрейм данных с двумя столбцами и иметь имена столбцов как «Col_A» и «Col_B».
Как я могу это сделать?>