Início / coding / Perguntas / 78501868
Accepted
Bogaso
Asked: 2024-05-19 15:25:17 +0800 CST

Defina o tipo de dados na criação de função personalizada

  • 772

Eu tenho abaixo a função personalizada

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

No entanto, quero forçar o usuário a fornecer um dataframe com 2 colunas e com nomes de colunas como'Col_A' and 'Col_B'

Qualquer ideia de como posso fazer isso seria muito apreciada.

python

2 respostas

  1. Aqui está o meu método

    Eu uso uma função try para evitar uma exceção, esta é uma maneira fácil de consertar e mais flexível

    import pandas as pd
def MyFn(DF : pd.DataFrame) -> float :
  try:
    return DF['Col_A'].values[1] - DF['Col_B'].values[1]
  except:
    print("you must include 'Col_A' and 'Col_B' in the argument!")
    return 0.0 # I would like to return something to prevent error

    Espero que isso resolva o problema.

    • 2
  2. Best Answer

    Você deve ser capaz de usar asserte DF.columns.tolist() == ['Col_A', 'Col_B']:

    def MyFn(DF : pd.DataFrame) -> float :
    assert DF.columns.tolist() == ['Col_A', 'Col_B'], 'DF must have two columns named "Col_A" and "Col_B"'
    return DF['Col_A'].values[1] - DF['Col_B'].values[1]

MyFn(pd.DataFrame([[1, 2], [3, 5]], columns=['Col_A', 'Col_B']))
# -2

MyFn(pd.DataFrame([[1, 2], [3, 5]], columns=['Col_A', 'Col_C']))
# AssertionError: DF must have two columns named "Col_A" and "Col_B"

MyFn(pd.DataFrame([[1, 2, 3], [4, 6, 7]], columns=['Col_A', 'Col_B', 'Col_C']))
# AssertionError: DF must have two columns named "Col_A" and "Col_B"

    Observe que isso exigiria exatamente duas colunas, "Col_A" e "Col_B", nesta ordem específica.

    Uma opção mais flexível seria usar try/ except, o que permitiria lidar com outro erro potencial de forma independente: o fato de passar um DataFrame com menos de duas linhas acionaria um IndexError:

    def MyFn(DF : pd.DataFrame) -> float :
    try:
        return DF['Col_A'].values[1] - DF['Col_B'].values[1]
    except KeyError as e:
        raise Exception('DF must contain "Col_A" and "Col_B"') from e
    except IndexError as e:
        raise Exception('DF must have at least two rows') from e

MyFn(pd.DataFrame([[1, 2], [3, 5]], columns=['Col_A', 'Col_B']))
# -2

#MyFn(pd.DataFrame([[1, 2], [3, 5]], columns=['Col_A', 'Col_C']))
# Exception: DF must contain "Col_A" and "Col_B"

MyFn(pd.DataFrame([[1, 2, 3], [4, 6, 7]], columns=['Col_A', 'Col_B', 'Col_C']))
# -2

MyFn(pd.DataFrame([[1, 2]], columns=['Col_A', 'Col_B']))
# Exception: DF must have at least two rows
    • 0

relate perguntas