Python の print の出力を「式: (計算結果)」となるようにしたい

Question

デバッグのための自作 print 関数

デバッグのために print 関数を拡張したいです。
このモジュールを使うと良いよ、というようなヒントがいただければと思います。

入出力

入力：数式
出力：「数式：（数式の計算結果）」

例1

以下の print() と同じ出力を、例えば my_print(w*x) とするだけで実現したいです。

import numpy as np

x = np.array([0, 1])
w = np.array([0.5, 0.5])

print('w*x: ' + str(w*x)) # --> w*x: [0.  0.5]

例2

例1と同様に、my_print(np.sum(w*x)) という形式で以下のコードと同じ出力を得たいです。

import numpy as np

x = np.array([0, 1])
w = np.array([0.5, 0.5])

print('np.sum(w*x): ' + str(np.sum(w*x))) # --> np.sum(w*x): 0.5

よろしくお願いいたします。

Comments

[melian]

コロンではなくてイコール(=) でもよければ f-strings が使えます(Python 3.8以降)。
print(f'{w*x = }')
print(f'{np.sum(w*x) = }')

[Ryuking885]

ありがとうございます！非常に助かりました。

Answer 1

お求めのものは、Pythonでは引数は関数に渡される前に評価されるので、そのままでは無理です。
少し危険ですがevalを使えば似たようなものは作れます。

>>> def my_print(s):
...     print(s+':', eval(s))
...
>>> import numpy as np
>>>
>>> x = np.array([0, 1])
>>> w = np.array([0.5, 0.5])
>>>
>>> my_print('w*x')
w*x: [0.  0.5]
>>>

Answer 2

Python3.8から標準でその機能があります。

a = 1
print(f'{a=}')

https://gammasoft.jp/blog/python-f-string/

python3.7以下なら

from inspect import currentframe

def chkprint(*args):
    names = {id(v):k for k,v in currentframe().f_back.f_locals.items()}
    print(', '.join(names.get(id(arg),'???')+' = '+repr(arg) for arg in args))

a = 1
chkprint(a)

ただし、int（-5~256）を引数に取ると正常に動作しません。