pythonで、ループ処理中に得る数値を、別のスクリプトで取得したい

Question

プログラミング初心者です。
python3でループ処理の関数の途中で得る数値を、別のスクリプトで取得したいのですが、やり方がわかりません。

*A.py
import time
x=0
def test():
  while True:
     x=x+1
     time.sleep(1)

*B.py
import time
import A
while True:
   print(A.test()) 
   time.sleep(1)

Ｂを動かすときに、ＡのＸを受けて、１ずつ出力結果が増えるようにしたいのですが、この場合、 Ａのtest()関数の戻り値をどう書けばいいのでしょうか？
どなたか教えていただけると幸いです。よろしくお願いします。

Comments

[Zuishin]

コードが間違っているので直すことになりますが、それによっては意図したものと違うものになります。どうなれば良いのかを詳しく書いてください。例えば私には A.py を無くすのが一番良い方法に見えます。また Python においてインデントは特別な意味を持ちます。それが全部消えているのでコードブロックを使って書き直してください。「teratail コードブロック」で検索したら多分出るんじゃないでしょうか。

[ros_tana]

インデントがなくて申し訳ありません。修正いたしました。ご指摘ありがとうございます。

Answer 1

ジェネレーター(=コルーチン)のことでしょうか・・・

import time

def test():
    x = 0
    while True:
        yield x
        x += 1

g = test()
while True:
    print(next(g))
    time.sleep(1)

==>
0
1
2
...

ちなみに、モジュールが同じかどうかは本件には無関係なのでtestは同一モジュールに定義しています。

必要とあらばtestでsleepを呼ぶこともできますが、そうすると2秒ごとにprintが実行されると思います。つまりコルーチンは呼出し元と並行に動作するわけではありません。nextの呼び出しでyieldで中断していたtestの実行が再開するだけだからです。

並行して動作させることを意図しておられるならコルーチンではなく独立したスレッドでtestを実行しつつ、なんらかの方法でスレッド間通信により中間結果(各xの値)を別のスレッドへ通知するような方法になると思います。

Comments

[ros_tana]

わかりやすいご回答ありがとうございます。
yieldを使うこと、またyieldの機能の仕方への、理解が進みました。
ご指摘の通り、並行して動作させたいと考えていたので、スレッド間通信が必要だとわかったのも、大きな前進でした。ありがとうございます。

Answer 2

a.testをジェネレータ関数にすれば良いかと。
Qiita - Pythonのイテレータとジェネレータ

---

b.py

import time
import a

def main():
    for elem in a.test():
        if elem > 10:
            break
        
        print(elem)
        # time.sleep(1)

if __name__ == '__main__':
    main()

a.py

def test():
    x = 0
    while True:
        x = x+1
        yield x

実行結果 Wandbox

1
2
3
4
5
6
7
8
9
10

---

なお、単にカウントアップする処理なら、itertools.countを使うと良いです。

from itertools import count
import time

for elem in count():
    if elem > 10:
        break

    print(elem)
    # time.sleep(1)

質問の仕方について

teratailには、上記のようにコードを見やすく表示する機能があります。
質問編集画面を開き、コードを選択した状態で<code>ボタンを押してください。
特にPythonの場合、インデントが崩れるとコードの意味が変わってしまいます。

Comments

[ros_tana]

早速ご回答いただきありがとうございます。yieldの使い方を教えていただき、学習が進みました。また、投稿の中でインデントをどう入れればいいかわからなかったので、わかりやすく教えていただき、無事に修正することができました。誠にありがとうございました！