我已经读了如何获得一个函数名作为字符串?。
如何对变量做同样的事情呢?与函数相反，Python变量没有__name__属性。
换句话说，如果我有一个变量，比如:
foo = dict()
foo['bar'] = 2
我正在寻找一个函数/属性，例如retrieve_name()，以便从这个列表中创建一个DataFrame in Pandas，其中列名由实际字典的名称给出:
# List of dictionaries for my DataFrame
list_of_dicts = [n_jobs, users, queues, priorities]
columns = [retrieve_name(d) for d in list_of_dicts]
当前回答
from __future__ import annotations
import inspect
import pandas as pd
# 변수 이름 가져오기
def getVariableName(variable: str) -> (str | Exception):
callers_local_vars = inspect.currentframe().f_back.f_locals.items()
df = pd.DataFrame(callers_local_vars)
df.columns = ['함수명', '값']
try:
return df[(df['값'] == variable)].iloc[0]['함수명']
except Exception as e:
return e
test = 'Hello World!'
if __name__ == '__main__':
print(getVariableName(test))
其他回答
这里有一种方法。我不建议在任何重要的事情上使用它，因为它会很脆。但这是可以做到的。
创建一个使用inspect模块查找调用它的源代码的函数。然后可以解析源代码，以识别想要检索的变量名。例如，这里有一个名为autodict的函数，它接受一个变量列表，并返回一个将变量名映射到变量值的字典。例如:
x = 'foo'
y = 'bar'
d = autodict(x, y)
print d
将:
{'x': 'foo', 'y': 'bar'}
检查源代码本身比搜索locals()或globals()更好，因为后一种方法不会告诉您哪些变量是您想要的。
无论如何，代码如下:
def autodict(*args):
get_rid_of = ['autodict(', ',', ')', '\n']
calling_code = inspect.getouterframes(inspect.currentframe())[1][4][0]
calling_code = calling_code[calling_code.index('autodict'):]
for garbage in get_rid_of:
calling_code = calling_code.replace(garbage, '')
var_names, var_values = calling_code.split(), args
dyn_dict = {var_name: var_value for var_name, var_value in
zip(var_names, var_values)}
return dyn_dict
该动作发生在inspect的行中。Getouterframes，它返回调用autodict的代码中的字符串。
这种魔法的明显缺点是，它对源代码的结构做出了假设。当然，如果它在解释器内部运行，它根本就不起作用。
您可以尝试以下方法来检索您定义的函数的名称(但不适用于内置函数):
import re
def retrieve_name(func):
return re.match("<function\s+(\w+)\s+at.*", str(func)).group(1)
def foo(x):
return x**2
print(retrieve_name(foo))
# foo
我试着从inspect locals中获取名称，但它不能处理var，比如[1]，b.val。 之后，我有了一个新的想法——从代码中获取var名称，我尝试了succ! 代码如下:
#direct get from called function code
def retrieve_name_ex(var):
stacks = inspect.stack()
try:
func = stacks[0].function
code = stacks[1].code_context[0]
s = code.index(func)
s = code.index("(", s + len(func)) + 1
e = code.index(")", s)
return code[s:e].strip()
except:
return ""
你可以以kwargs的形式获取变量，并以字符串的形式返回:
var=2
def getVarName(**kwargs):
return list(kwargs.keys())[0]
print (getVarName(var = var))
注:变量名必须等于自身。
from __future__ import annotations
import inspect
import pandas as pd
# 변수 이름 가져오기
def getVariableName(variable: str) -> (str | Exception):
callers_local_vars = inspect.currentframe().f_back.f_locals.items()
df = pd.DataFrame(callers_local_vars)
df.columns = ['함수명', '값']
try:
return df[(df['값'] == variable)].iloc[0]['함수명']
except Exception as e:
return e
test = 'Hello World!'
if __name__ == '__main__':
print(getVariableName(test))
