Abaikan nilai balik multipel python

Katakanlah saya memiliki fungsi Python yang mengembalikan beberapa nilai dalam sebuah tuple:

def func():
    return 1, 2

Apakah ada cara yang baik untuk mengabaikan salah satu hasil daripada hanya menetapkan ke variabel sementara? Katakanlah jika saya hanya tertarik pada nilai pertama, apakah ada cara yang lebih baik dari ini:

x, temp = func()

Satu konvensi yang umum adalah menggunakan "_" sebagai nama variabel untuk elemen tuple yang ingin Anda abaikan. Misalnya:

def f():
    return 1, 2, 3

_, _, x = f()

Anda dapat menggunakan x = func()[0]untuk mengembalikan nilai pertama, x = func()[1]untuk mengembalikan nilai kedua, dan seterusnya.

Jika Anda ingin mendapatkan beberapa nilai sekaligus, gunakan sesuatu seperti x, y = func()[2:4].

Jika Anda menggunakan Python 3, Anda dapat menggunakan bintang sebelum variabel (di sebelah kiri tugas) untuk menjadikannya daftar pembongkaran.

# 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]

Ingat, ketika Anda mengembalikan lebih dari satu item, Anda benar-benar mengembalikan tuple. Jadi Anda dapat melakukan hal-hal seperti ini:

def func():
    return 1, 2

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

Praktik umum adalah menggunakan variabel dummy _(underscore tunggal), seperti yang telah ditunjukkan banyak orang di sini sebelumnya.

Namun, untuk menghindari tabrakan dengan penggunaan lain dari nama variabel itu (lihat respons ini ) mungkin lebih baik menggunakan __(menggandakan garis bawah) daripada sebagai variabel sekali pakai, seperti yang ditunjukkan oleh ncoghlan . Misalnya:

x, __ = func()

Tiga pilihan sederhana.

Jelas

x, _ = func()

x, junk = func()

Mengerikan

x = func()[0]

Dan ada cara untuk melakukan ini dengan dekorator.

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

func0= val0(func)

Solusi terbaik mungkin adalah memberi nama sesuatu daripada mengembalikan tuple yang tidak berarti. Kecuali ada logika di balik urutan barang yang dikembalikan.

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

latitude = func()['lat']

Anda bahkan dapat menggunakan namedtuple jika Anda ingin menambahkan informasi tambahan tentang apa yang Anda kembalikan (ini bukan hanya kamus, ini adalah koordinat):

from collections import namedtuple 

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

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

latitude = func().lat

Jika objek dalam kamus / tuple Anda sangat terikat bersama-sama maka mungkin ide yang baik untuk bahkan menentukan kelas untuk itu. Dengan begitu Anda juga akan dapat mendefinisikan interaksi antara objek-objek jenis ini dan memberikan API tentang cara bekerja dengannya. Pertanyaan alami yang mengikuti: Kapan saya harus menggunakan kelas dengan Python? .

Ini sepertinya pilihan terbaik bagi saya:

val1, val2, ignored1, ignored2 = some_function()

Ini tidak samar atau jelek (seperti metode func () [index]), dan dengan jelas menyatakan tujuan Anda.

Ini bukan jawaban langsung untuk pertanyaan itu. Melainkan menjawab pertanyaan ini: "Bagaimana saya memilih output fungsi spesifik dari banyak opsi yang mungkin?".

Jika Anda dapat menulis fungsi (mis., Itu tidak ada di perpustakaan yang tidak dapat Anda modifikasi), kemudian tambahkan argumen input yang menunjukkan apa yang Anda inginkan dari fungsi tersebut. Buatlah argumen bernama dengan nilai default sehingga dalam "kasus umum" Anda bahkan tidak harus menentukannya.

    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 )

Metode ini memberikan fungsi "peringatan lanjutan" tentang output yang diinginkan. Akibatnya ia dapat melewati pemrosesan yang tidak dibutuhkan dan hanya melakukan pekerjaan yang diperlukan untuk mendapatkan hasil yang diinginkan. Juga karena Python melakukan pengetikan dinamis, tipe pengembalian dapat berubah. Perhatikan bagaimana contoh mengembalikan skalar, daftar atau tuple ... apa pun yang Anda suka!

Jika ini adalah fungsi yang Anda gunakan sepanjang waktu tetapi selalu membuang argumen kedua, saya berpendapat bahwa itu kurang berantakan untuk membuat alias untuk fungsi tanpa menggunakan nilai balik kedua menggunakan lambda.

def func():
    return 1, 2

func_ = lambda: func()[0] 

func_()  # Prints 1 

Ketika Anda memiliki banyak output dari suatu fungsi dan Anda tidak ingin menyebutnya berkali-kali, saya pikir cara paling jelas untuk memilih hasilnya adalah:

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

Sebagai contoh kerja minimum, juga menunjukkan bahwa fungsi ini dipanggil hanya sekali:

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]]

Dan nilainya seperti yang diharapkan:

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

Untuk kenyamanan, indeks daftar Python juga dapat digunakan:

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

Pengembalian: a = 3 dan b = 12