Zignoruj ​​wielokrotne zwracane wartości w Pythonie

Powiedzmy, że mam funkcję Python, która zwraca wiele wartości w krotce:

def func():
    return 1, 2

Czy istnieje dobry sposób na zignorowanie jednego z wyników zamiast przypisywania go do zmiennej tymczasowej? Powiedz, że gdybym był zainteresowany tylko pierwszą wartością, czy istnieje lepszy sposób niż ten:

x, temp = func()

Jedną z powszechnych konwencji jest użycie „_” jako nazwy zmiennej dla elementów krotki, które chcesz zignorować. Na przykład:

def f():
    return 1, 2, 3

_, _, x = f()

Możesz użyć x = func()[0]do zwrócenia pierwszej wartości, x = func()[1]do zwrócenia drugiej itd.

Jeśli chcesz uzyskać wiele wartości jednocześnie, użyj czegoś takiego x, y = func()[2:4].

Jeśli używasz Python 3, możesz użyć gwiazdki przed zmienną (po lewej stronie przypisania), aby była to lista podczas rozpakowywania.

# Example 1: a is 1 and b is [2, 3]

a, *b = [1, 2, 3]

# Example 2: a is 1, b is [2, 3], and c is 4

a, *b, c = [1, 2, 3, 4]

# Example 3: b is [1, 2] and c is 3

*b, c = [1, 2, 3]       

# Example 4: a is 1 and b is []

a, *b = [1]

Pamiętaj, że kiedy zwracasz więcej niż jeden przedmiot, naprawdę zwracasz krotkę. Możesz więc robić takie rzeczy:

def func():
    return 1, 2

print func()[0] # prints 1
print func()[1] # prints 2

Powszechną praktyką jest używanie zmiennej fikcyjnej _ (pojedynczy podkreślnik), jak wiele osób wskazało tutaj wcześniej.

Jednak, aby uniknąć kolizji z innymi zastosowaniami tej nazwy zmiennej (patrz ta odpowiedź), lepszym rozwiązaniem może być użycie __(podwójnego podkreślenia) zamiast zmiennej odchodzącej , jak wskazuje ncoghlan . Na przykład:

x, __ = func()

Trzy proste opcje.

Oczywisty

x, _ = func()

x, junk = func()

Ohydny

x = func()[0]

I są sposoby, aby to zrobić za pomocą dekoratora.

def val0( aFunc ):
    def pick0( *args, **kw ):
        return aFunc(*args,**kw)[0]
    return pick0

func0= val0(func)

Prawdopodobnie najlepszym rozwiązaniem jest nazywanie rzeczy zamiast zwracania bezsensownych krotek. Chyba że za kolejnością zwracanych przedmiotów kryje się jakaś logika.

def func():
    return {'lat': 1, 'lng': 2}

latitude = func()['lat']

Możesz nawet użyć namedtuple, jeśli chcesz dodać dodatkowe informacje o tym, co zwracasz (to nie jest tylko słownik, to są współrzędne):

from collections import namedtuple 

Coordinates = namedtuple('Coordinates', ['lat', 'lng'])

def func():
    return Coordinates(lat=1, lng=2)

latitude = func().lat

Jeśli obiekty w twoim słowniku / krotce są ściśle ze sobą powiązane, dobrym pomysłem może być nawet zdefiniowanie dla niej klasy. W ten sposób będziesz mógł zdefiniować interakcję między obiektami tego typu i podać interfejs API, jak z nimi pracować. Naturalne pytanie: Kiedy powinienem używać klas w Pythonie? .

To wydaje mi się najlepszym wyborem:

val1, val2, ignored1, ignored2 = some_function()

Nie jest tajemniczy ani brzydki (jak metoda func () [index]) i wyraźnie określa twój cel.

To nie jest bezpośrednia odpowiedź na pytanie. Raczej odpowiada na to pytanie: „Jak wybrać konkretną funkcję z wielu możliwych opcji?”.

Jeśli jesteś w stanie napisać funkcję (tzn. Nie ma jej w bibliotece, której nie można zmodyfikować), dodaj argument wejściowy wskazujący, czego chcesz od funkcji. Zrób z niej nazwany argument o wartości domyślnej, aby w „zwykłym przypadku” nawet nie musieć go podawać.

    def fancy_function( arg1, arg2, return_type=1 ):
        ret_val = None
        if( 1 == return_type ):
            ret_val = arg1 + arg2
        elif( 2 == return_type ):
            ret_val = [ arg1, arg2, arg1 * arg2 ]
        else:
            ret_val = ( arg1, arg2, arg1 + arg2, arg1 * arg2 ) 
        return( ret_val )

Ta metoda daje funkcji „zaawansowane ostrzeżenie” dotyczące pożądanego wyjścia. W rezultacie może ominąć niepotrzebne przetwarzanie i wykonać tylko pracę niezbędną do uzyskania pożądanego wyniku. Ponieważ Python wykonuje dynamiczne pisanie, typ zwracany może się zmienić. Zauważ, jak przykład zwraca skalar, listę lub krotkę ... cokolwiek chcesz!

Jeśli jest to funkcja, której używasz cały czas, ale zawsze odrzucasz drugi argument, argumentowałbym, że mniej bałagan jest tworzenie aliasu dla funkcji bez użycia drugiej wartości zwracanej lambda.

def func():
    return 1, 2

func_ = lambda: func()[0] 

func_()  # Prints 1 

Jeśli masz wiele danych wyjściowych z funkcji i nie chcesz wywoływać jej wiele razy, myślę, że najlepszym sposobem na wybranie wyników byłoby:

results = fct()
a,b = [results[i] for i in list_of_index]

Jako minimum działający przykład, pokazujący, że funkcja jest wywoływana tylko raz:

def fct(a):
    b=a*2
    c=a+2
    d=a+b
    e=b*2
    f=a*a
    print("fct called")
    return[a,b,c,d,e,f]

results=fct(3)
> fct called

x,y = [results[i] for i in [1,4]]

A wartości są zgodne z oczekiwaniami:

results
> [3,6,5,9,12,9]
x
> 6
y
> 12

Dla wygody można również użyć indeksów list w języku Python:

x,y = [results[i] for i in [0,-2]]

Zwraca: a = 3 ib = 12