Python：return文の後に自動的にクラスオブジェクトを削除する

Question

私はHDF5ファイルから読み込んだ潜在的に非常に大きな凹凸配列を作成して格納する関数を含むクラスを持っています。 （私はクラスオブジェクトとして配列を格納するので、ファイルからさまざまなHDF5データセットを読み込むと、手動ループではなくリストの理解が可能になります）。私が望むのは、格納された配列を同時に返したり削除したりすることですreturn文の後にメモリを置く）。

class MWE(object): 
     def __init__(self,*args): 
      pass 
      return 

     def example(self,**kwargs): 
      self.array = readLargeArrayFromHDF5File(...) 
      return self.array # And simultaneously clean up self.array? 

私は単に私のスクリプトの後半でdel MWE.arrayのようなものを使用してアレイを削除することができることを知っているが、自動的に私はこれを行うには覚えていなくても、クリーンアップする方法はありますか？

ありがとうございます！

Answer 1

あなたはこの種のもののためにcontext managerを使用することにネーテッド！ withコードブロックからオブジェクトが自動的にクリアされます。

from contextlib import contextmanager 


class MWE(object): 

    def __init__(self, *args): 
     pass 
     return 

    @contextmanager 
    def example(self, **kwargs): 
     array = readLargeArrayFromHDF5File(...) 
     try: 
      yield array 
     finally: 
      array = None 

    @contextmanager 
    def example_two(self, **kwargs): 
     array = readLargeArrayFromHDF5File(...) 
     try: 
      for item in array: 
       yield item 
     finally: 
      array = None 
# usage 
with mwe_obj.example_two(kwargs) as result: 
    for r in result: 
     # work with your result here 
     print (r) 

# when your code hits this block self.array would get set to None 
print ('cleared large array')