Início / coding / Perguntas / 78934295
Accepted
Paul Jurczak
Asked: 2024-08-31 12:13:26 +0800 CST

Essas duas funções Python podem ser substituídas por uma única função genérica que aceite um argumento de lista ou de tupla?

  • 772

Essas duas funções Python podem:

def negativeList(a):
  return [-e for e in a]

def negativeTuple(a):
  return tuple(-e for e in a)

ser substituído por uma única função genérica equivalente negative(a)?

python

4 respostas

  1. Você pode usar a função isinstance para verificar o tipo de entrada e então retornar o tipo apropriado.

    def negative_sequence(a):
    if isinstance(a, list):
        return [-e for e in a]
    elif isinstance(a, tuple):
        return tuple(-e for e in a)
    else:
        raise TypeError("Input must be a list or a tuple")

# Example usage:
print(negative_sequence([1, 2, 3]))  # Output: [-1, -2, -3]
print(negative_sequence((1, 2, 3)))  # Output: (-1, -2, -3)
    • 5
  2. Best Answer

    Você pode determinar o tipo em tempo de execução

    def negative(a):
    return type(a)(-e for e in a)
    • 4

  3. Então também podemos usar a Iterableclasse tornando-a mais robusta

    from collections.abc import Iterable
import numpy as np


def negative(a):
    if isinstance(a, Iterable):
        result = (-e for e in a)
        if isinstance(a, list):
            return list(result)
        elif isinstance(a, tuple):
            return tuple(result)
        elif isinstance(a, np.ndarray):
            return np.negative(a)
        else:
            raise TypeError("Not supported iterable type.")
    else:
        raise TypeError("Interables only.")
    • 0

  4. Acredito que é para isso que os decoradores de despacho único podem ser usados:

    from functools import singledispatch

@singledispatch
def negative(arg):
    return f'unsupported type: {type(arg)}'

@negative.register
def _(arg: tuple):
    return tuple(-e for e in arg)

@negative.register
def _(arg: list):
    return list(-e for e in arg

print(negative(10))
print(negative('abc'))
print(negative([1, 2, 4]))
print(negative((1, 2, 3)))

    Notas:

    • Eu preferiria gerar uma exceção do que retornar uma string de erro. Essa parte é apenas para facilitar a demonstração.
    • Você pode diferenciar ainda mais usando eg list[int]ou tuple[float], a documentação vinculada tem exemplos lá e uma pesquisa na web deve revelar muito mais.
    • 0

relate perguntas