欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python打印变量名

程序员文章站 2023-12-30 16:57:22
...
aa = 1
print(aa)

输出结果:

1

使用print简单打印变量的时候,只会显示变量名对应的值,并不会打印变量名,那如何才能打印变量名呢?

python中内置locals()是打印变量名的关键。

aa = 1
print(locals())

输出结果:

{..., 'aa': 1, '__name__': '__main__', '__spec__': None,...}  

可以看到变量aa的名字被保存在此字典中,所以我们可以通过逐一检查字典的值是否和变量aa指向的值是同一个值来得到变量名,话不多说,上代码。

def var_name(var,all_var=locals()):
    return [var_name for var_name in all_var if all_var[var_name] is var][0]

aa = 11
bb = 22
cc = 33
dd = 44
print(var_name(aa))
print(var_name(bb))
print(var_name(cc))
print(var_name(dd))

输出结果:

aa
bb
cc
dd

在函数头里面使用locals()作为默认参数,得到的字典是外部的局部变量参数字典(新发现,嘿嘿嘿)。

但是这样做会存在问题,那就是当多个变量名指向同一个值的时候就会发错名字混乱。

def var_name(var, all_var=locals()):
    return [var_name for var_name in all_var if all_var[var_name] is var][0]


aa = bb = 11
cc = dd = 22
print(var_name(aa))
print(var_name(bb))
print(var_name(cc))
print(var_name(dd))

输出结果:

bb
bb
cc
cc

所以只能当变量的值只被引用一次的时候才可以这么用。

终极解决方法

在*上找到的答案

import inspect, re

def varname(p):
  for line in inspect.getframeinfo(inspect.currentframe().f_back)[3]:
    m = re.search(r'\bvarname\s*\(\s*([A-Za-z_][A-Za-z0-9_]*)\s*\)', line)
    if m:
      return m.group(1)


spam = 42
print(varname(spam))

输出结果:

spam

原理:通过检查调用varname所在行的代码的字符串得到变量名。

参考资料:
*上的解决方法

上一篇:

下一篇: