def文の戻り値を、そのdef文の引数に繰り返し入れる方法

Question

前提・実現したいこと

当方pythonを始めて3か月の初心者です。初めての質問です。お手柔らかにお願いします。質問の意図が分かりづらかったら申し訳ありません。

def文で定義した関数A（この時の引数をx0とする）の戻り値（これをx1とする）を、関数Aの引数（x1）として、その戻り値（同様に、x2とする）を関数Aの引数（x2）とする...
この処理を20回ほど繰り返し行いたいのですが、そのままプログラムを書いてしまうと、とても長くなってしまうので、for文やその他の関数を使って繰り返し処理を行いたいです。なお、引数、戻り値ともに1次元のリストになっています。
よろしくお願いいたします。

以下のようなプログラムを繰り返し処理を使って短縮したいです。
以下のプログラムはあくまでも例なので、見苦しいプログラムかと思いますが、ご容赦ください。

import numpy as np
sequence=np.random.rand(21)
def A(x0):
    X1=np.zeros(21)
    X_1=[]
    for i in range(1,20):
        x1=x0[i-1]/4+x0[i]/2+x0[i+1]/4
        X_1=np.append(X_1,x1)
    X1_end=x0[19]/4+x0[20]/2
    X1[1:20]=X_1
    X1[0]=0
    X1[20]=X1_end
    return X1
X1=A(sequence)
X2=A(X1)
X3=A(X2)
X4=A(X3)
以下、X20となるまで繰り返し
X2=A(X1)以降の処理を繰り返し処理などで短縮したい。

Answer 1

X = sequence
for i in range(20):
    X = A(X)

---

補記……A()はnp.convolve()を用いることで以下のように短くできます。

def A(x0):
    x1 = np.convolve(x0, np.array([1/4, 1/2, 1/4]), mode='same')
    x1[0] = 0
    return x1

Comments

[shosin]

無事解決できました。ありがとうございました。

Answer 2

return X1

を、A(X1) とすればいいです。
が、単純にこれすると延々と無限ループしてしまいますんで、終了の条件を設定して終わる必要があります

Comments

[shosin]

迅速な回答をありがとうございます。
ちなみに終了の条件設定とはどのようにプログラムを入力すればよろしいでしょうか。
お手数をおかけしますがよろしくお願いいたします。

[y_waiwai]

繰り返し回数だけなら、グローバル変数で変数用意して、関数実行するたびに＋１していって、想定回数以上になるとreturn するようにすればよろしい

[shosin]

無事解決できました。ありがとうございました。