Bagaimana cara memeriksa apakah suatu objek adalah objek generator di python?

Dengan python, bagaimana cara memeriksa apakah suatu objek adalah objek generator?

Mencoba ini -

>>> type(myobject, generator)

memberikan kesalahan -

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'generator' is not defined

(Saya tahu saya dapat memeriksa apakah objek memiliki nextmetode untuk menjadi generator, tetapi saya ingin beberapa cara menggunakan yang saya dapat menentukan jenis objek apa pun, bukan hanya generator.)

Anda dapat menggunakan GeneratorType dari jenis:

>>> import types
>>> types.GeneratorType
<class 'generator'>
>>> gen = (i for i in range(10))
>>> isinstance(gen, types.GeneratorType)
True

Maksud Anda fungsi generator? gunakan inspect.isgeneratorfunction.

EDIT:

jika Anda menginginkan objek generator, Anda dapat menggunakan inspect.isgenerator seperti yang ditunjukkan oleh JAB dalam komentarnya.

Saya pikir penting untuk membuat perbedaan antara fungsi generator dan generator (hasil fungsi generator):

>>> def generator_function():
...     yield 1
...     yield 2
...
>>> import inspect
>>> inspect.isgeneratorfunction(generator_function)
True

memanggil generator_function tidak akan menghasilkan hasil normal, bahkan tidak akan mengeksekusi kode apa pun dalam fungsi itu sendiri, hasilnya akan menjadi objek khusus yang disebut generator :

>>> generator = generator_function()
>>> generator
<generator object generator_function at 0x10b3f2b90>

jadi ini bukan fungsi generator, tapi generator:

>>> inspect.isgeneratorfunction(generator)
False

>>> import types
>>> isinstance(generator, types.GeneratorType)
True

dan fungsi generator bukan generator:

>>> isinstance(generator_function, types.GeneratorType)
False

hanya untuk referensi, panggilan fungsi tubuh yang sebenarnya akan terjadi dengan mengkonsumsi generator, misalnya:

>>> list(generator)
[1, 2]

Lihat juga Dalam python apakah ada cara untuk memeriksa apakah suatu fungsi merupakan "fungsi generator" sebelum memanggilnya?

The inspect.isgeneratorFungsi baik-baik saja jika Anda ingin memeriksa generator murni (yaitu objek dari "generator" kelas). Namun itu akan kembali Falsejika Anda memeriksa, misalnya, izipiterable. Cara alternatif untuk memeriksa generator yang umum adalah dengan menggunakan fungsi ini:

def isgenerator(iterable):
    return hasattr(iterable,'__iter__') and not hasattr(iterable,'__len__')

Anda bisa menggunakan Iterator atau lebih spesifik, Generator dari modul pengetikan .

from typing import Generator, Iterator
g = (i for i in range(1_000_000))
print(type(g))
print(isinstance(g, Generator))
print(isinstance(g, Iterator))

hasil:

<class 'generator'>
True
True

>>> import inspect
>>> 
>>> def foo():
...   yield 'foo'
... 
>>> print inspect.isgeneratorfunction(foo)
True

Saya tahu saya dapat memeriksa apakah objek memiliki metode berikutnya untuk menjadi generator, tetapi saya ingin beberapa cara menggunakan yang saya dapat menentukan jenis objek apa pun, bukan hanya generator.

Jangan lakukan ini. Itu hanya ide yang sangat, sangat buruk.

Sebaliknya, lakukan ini:

try:
    # Attempt to see if you have an iterable object.
    for i in some_thing_which_may_be_a_generator:
        # The real work on `i`
except TypeError:
     # some_thing_which_may_be_a_generator isn't actually a generator
     # do something else

Dalam hal tidak mungkin bahwa tubuh for for loop juga memiliki TypeErrors, ada beberapa pilihan: (1) menentukan fungsi untuk membatasi ruang lingkup kesalahan, atau (2) menggunakan blok percobaan bersarang .

Atau (3) sesuatu seperti ini untuk membedakan semua ini TypeErroryang beredar.

try:
    # Attempt to see if you have an iterable object.
    # In the case of a generator or iterator iter simply 
    # returns the value it was passed.
    iterator = iter(some_thing_which_may_be_a_generator)
except TypeError:
     # some_thing_which_may_be_a_generator isn't actually a generator
     # do something else
else:
    for i in iterator:
         # the real work on `i`

Atau (4) perbaiki bagian lain dari aplikasi Anda untuk menyediakan generator dengan tepat. Itu seringkali lebih sederhana dari semua ini.

Jika Anda menggunakan server web tornado atau sejenisnya, Anda mungkin menemukan bahwa metode server sebenarnya adalah generator dan bukan metode. Ini membuatnya sulit untuk memanggil metode lain karena hasil tidak bekerja di dalam metode dan karena itu Anda perlu mulai mengelola kumpulan objek generator yang dirantai. Metode sederhana untuk mengelola kumpulan generator berantai adalah dengan membuat fungsi bantuan seperti

def chainPool(*arg):
    for f in arg:
      if(hasattr(f,"__iter__")):
          for e in f:
             yield e
      else:
         yield f

Sekarang menulis generator berantai seperti

[x for x in chainPool(chainPool(1,2),3,4,chainPool(5,chainPool(6)))]

Menghasilkan output

[1, 2, 3, 4, 5, 6]

Yang mungkin adalah apa yang Anda inginkan jika Anda ingin menggunakan generator sebagai alternatif atau sejenisnya.

(Saya tahu ini adalah posting lama.) Tidak perlu mengimpor modul, Anda dapat mendeklarasikan objek untuk perbandingan di awal program:

gentyp= type(1 for i in "")                                                                                          
       ...
type(myobject) == gentyp