python打印变量名

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

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

终极解决方法

在stackoverflow上找到的答案

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所在行的代码的字符串得到变量名。

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

你可能感兴趣的:(python打印变量名)