Pythonでfor文の中に自作の関数は定義できる？パフォーマンスやメモリの観点からはどう？

2023年5月23日

今回はPythonでfor文の中に自作の関数は定義することができるのかについてご紹介いたします。

Pythonでfor文の中に自作の関数は定義できる？パフォーマンスやメモリの観点からはどう？

結論から言うとPythonの仕組みとしては可能です。

表題のことが何のことを表しているかというと、下記のようにfor文の中に自作の関数を定義できるのか、定義したとしてパフォーマンスが悪くなったりメモリ的に非効率になるのかというのが今回の議題となります。

for elem1 in mylist:
    def function1(elem2):
        print(elem2 +'test')
    function1(elem1)

for elem1 in mylist:

def function1(elem2):

print(elem2 +'test')

function1(elem1)

結論から言ってこれは問題ないです。
関数をfor文の外に書いたsample1.pyとfor文の中に書いたsample2.pyを用意し、各コードでAtCoder上で実行したところ、メモリとパフォーマンスで違いはほとんど現れませんでした。

mylist = []
for i in range(1000000):
  mylist.append([i])

def function1(elem1):
    for elem2 in elem1:
      pass

for elem1 in mylist:
    function1(elem1)

mylist = []

for i in range(1000000):

mylist.append([i])

def function1(elem1):

for elem2 in elem1:

pass

for elem1 in mylist:

function1(elem1)

終了コード 0
実行時間 787 ms
メモリ 122144 KB

mylist = []
for i in range(1000000):
  mylist.append([i])

for elem1 in mylist:
    def function1():
        for elem2 in elem1:
          pass
    function1()