繁体   English   中英

创建单独变量字典的更简单方法?

[英]Simpler way to create dictionary of separate variables?

我希望能够以字符串的形式获取变量的名称,但我不知道 Python 是否具有那么多的内省功能。 就像是:

>>> print(my_var.__name__)
'my_var'

我想这样做是因为我有一堆变量,我想变成像这样的字典:

bar = True
foo = False
>>> my_dict = dict(bar=bar, foo=foo)
>>> print my_dict 
{'foo': False, 'bar': True}

但我想要比这更自动的东西。

Python 有locals()vars() ,所以我想有一种方法。

正如 unwind 所说,这实际上并不是您在 Python 中所做的事情 - 变量实际上是到对象的名称映射。

但是,这是尝试执行此操作的一种方法:

 >>> a = 1
 >>> for k, v in list(locals().iteritems()):
         if v is a:
             a_as_str = k
 >>> a_as_str
 a
 >>> type(a_as_str)
 'str'

我很想这样做。 这个 hack 与 rlotun 的建议非常相似,但它是单行的,这对我来说很重要:

blah = 1
blah_name = [ k for k,v in locals().iteritems() if v is blah][0]

蟒蛇 3+

blah = 1
blah_name = [ k for k,v in locals().items() if v is blah][0]

你想这样做吗?

dict( (name,eval(name)) for name in ['some','list','of','vars'] )

例子

>>> some= 1
>>> list= 2
>>> of= 3
>>> vars= 4
>>> dict( (name,eval(name)) for name in ['some','list','of','vars'] )
{'list': 2, 'some': 1, 'vars': 4, 'of': 3}

这是一个黑客。 它不适用于所有 Python 实现发行版(特别是那些没有traceback.extract_stack 。)

import traceback

def make_dict(*expr):
    (filename,line_number,function_name,text)=traceback.extract_stack()[-2]
    begin=text.find('make_dict(')+len('make_dict(')
    end=text.find(')',begin)
    text=[name.strip() for name in text[begin:end].split(',')]
    return dict(zip(text,expr))

bar=True
foo=False
print(make_dict(bar,foo))
# {'foo': False, 'bar': True}

请注意,此 hack 很脆弱:

make_dict(bar,
          foo)

(在 2 行上调用 make_dict)将不起作用。

与其尝试从foobar中生成 dict ,不如从字符串变量名称'foo''bar'生成 dict 更加 Pythonic :

dict([(name,locals()[name]) for name in ('foo','bar')])

这在 Python 中是不可能的,它真的没有“变量”。 Python 有名字,同一个对象可以有多个名字。

我认为我的问题将有助于说明为什么这个问题是有用的,并且它可能会更深入地了解如何回答它。 我编写了一个小函数来对代码中的各种变量进行快速内联头部检查。 基本上,它列出了变量名称、数据类型、大小和其他属性,因此我可以快速发现我犯的任何错误。 代码很简单:

def details(val):
  vn = val.__name__                 #  If such a thing existed
  vs = str(val)
  print("The Value of "+ str(vn) + " is " + vs)
  print("The data type of " + vn + " is " + str(type(val)))

所以如果你有一些复杂的字典/列表/元组情况,让解释器返回你分配的变量名会很有帮助。 例如,这是一个奇怪的字典:

m = 'abracadabra'
mm=[]    
for n in m:
  mm.append(n)
mydic = {'first':(0,1,2,3,4,5,6),'second':mm,'third':np.arange(0.,10)}



details(mydic)

The Value of mydic is {'second': ['a', 'b', 'r', 'a', 'c', 'a', 'd', 'a', 'b', 'r', 'a'], 'third': array([ 0.,  1.,  2.,  3.,  4.,  5.,  6.,  7.,  8.,  9.]), 'first': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]}
The data type of mydic is <type 'dict'>

details(mydic['first'])
The Value of mydic['first'] is (0, 1, 2, 3, 4, 5, 6)]
The data type of mydic['first'] is <type 'list'>

details(mydic.keys())
The Value of mydic.keys() is ['second', 'third', 'first']
The data type of mydic.keys() is <type 'tuple'>

details(mydic['second'][0])
The Value of mydic['second'][0] is a
The data type of mydic['second'][0] is <type 'str'>

我不确定我是否把它放在正确的地方,但我认为它可能会有所帮助。 我希望它会。

我根据这个问题的答案写了一个整洁的小有用函数。 我把它放在这里以防它有用。

def what(obj, callingLocals=locals()):
    """
    quick function to print name of input and value. 
    If not for the default-Valued callingLocals, the function would always
    get the name as "obj", which is not what I want.    
    """
    for k, v in list(callingLocals.items()):
         if v is obj:
            name = k
    print(name, "=", obj)

用法:

>> a = 4
>> what(a)
a = 4
>>|

我发现如果您已经有一个特定的值列表,那么@S 描述的方式。 洛茨是最好的; 但是,下面描述的方法可以很好地在整个代码中添加所有变量和类,而无需提供变量名称,尽管您可以根据需要指定它们。 代码可以扩展以排除类。

import types
import math  # mainly showing that you could import what you will before d

# Everything after this counts
d = dict(globals())

def kv_test(k,v):
    return (k not in d and 
            k not in ['d','args'] and
            type(v) is not types.FunctionType)

def magic_print(*args):
    if len(args) == 0: 
        return {k:v for k,v in globals().iteritems() if kv_test(k,v)}
    else:
        return {k:v for k,v in magic_print().iteritems() if k in args}

if __name__ == '__main__':
    foo = 1
    bar = 2
    baz = 3
    print magic_print()
    print magic_print('foo')
    print magic_print('foo','bar')

输出:

{'baz': 3, 'foo': 1, 'bar': 2}
{'foo': 1}
{'foo': 1, 'bar': 2}

在python 3中这很容易

myVariable = 5
for v in locals():
  if id(v) == id("myVariable"):
    print(v, locals()[v])

这将打印:

我的变量 5

蟒蛇3。 使用检查来捕获调用本地命名空间,然后使用这里介绍的想法。 正如所指出的,可以返回多个答案。

def varname(var):
  import inspect
  frame = inspect.currentframe()
  var_id = id(var)
  for name in frame.f_back.f_locals.keys():
    try:
      if id(eval(name)) == var_id:
        return(name)
    except:
      pass

这是我创建的用于读取变量名称的函数。 它更通用,可用于不同的应用程序:

def get_variable_name(*variable):
    '''gets string of variable name
    inputs
        variable (str)
    returns
        string
    '''
    if len(variable) != 1:
        raise Exception('len of variables inputed must be 1')
    try:
        return [k for k, v in locals().items() if v is variable[0]][0]
    except:
        return [k for k, v in globals().items() if v is variable[0]][0]

要在指定的问题中使用它:

>>> foo = False
>>> bar = True
>>> my_dict = {get_variable_name(foo):foo, 
               get_variable_name(bar):bar}
>>> my_dict
{'bar': True, 'foo': False}

在阅读线程时,我看到了非常多的摩擦。 给出一个错误的答案很容易,然后让某人给出正确的答案。 无论如何,这就是我发现的。

来自:[effbot.org]( http://effbot.org/zone/python-objects.htm#names

, so that was the clue.注意:它说,所以这就是线索。 Python 对象不是自引用的。 .然后它说, 我们在 TCL/TK 中有这个。 所以也许我的回答会有所帮助(但它确实帮助了我)

\n\n     jj = 123\n    打印 eval("'" + str(id(jj)) + "'")\n    打印目录()\n\n

\n 166707048\n ['__builtins__', '__doc__', '__file__', '__name__', '__package__', 'jj']\n

所以列表末尾有'jj'。

将代码改写为:

\n\n     jj = 123\n    打印 eval("'" + str(id(jj)) + "'")\n    对于 dir() 中的 x:\n        打印 id(eval(x))\n\n
\n 161922920\n ['__builtins__', '__doc__', '__file__', '__name__', '__package__', 'jj']\n 3077447796\n 136515736\n 3077408320\n 3077656800\n 136515736\n 161922920\n

这个讨厌的代码 id 是变量/对象/任何你学究调用它的名称。

所以,就是这样。 'jj' 的内存地址在我们直接查找时是一样的,就像我们在全局名称空间中进行字典查找时一样。 我相信你可以制作一个函数来做到这一点。 只需记住您的变量/对象/wypci 所在的命名空间。

QED。

我写了包法术来强大地做这种魔法。 你可以写:

from sorcery import dict_of

my_dict = dict_of(foo, bar)

也许我想多了,但是..

str_l = next((k for k,v in locals().items() if id(l) == id(v)))


>>> bar = True
>>> foo = False
>>> my_dict=dict(bar=bar, foo=foo)
>>> next((k for k,v in locals().items() if id(bar) == id(v)))
'bar'
>>> next((k for k,v in locals().items() if id(foo) == id(v)))
'foo'
>>> next((k for k,v in locals().items() if id(my_dict) == id(v)))
'my_dict'
import re
import traceback

pattren = re.compile(r'[\W+\w+]*get_variable_name\((\w+)\)')
def get_variable_name(x):
    return pattren.match( traceback.extract_stack(limit=2)[0][3]) .group(1)

a = 1
b = a
c = b
print get_variable_name(a)
print get_variable_name(b)
print get_variable_name(c)

我上传了一个解决方案pypi 它是一个定义等效于 C# 的nameof函数的模块。

它遍历其调用的帧的字节码指令,获取传递给它的变量/属性的名称。 这些名称位于函数名称.argreprLOAD指令的.argrepr中。

好吧,几天前我遇到了同样的需求不得不获得一个指向对象本身的变量

为什么如此必要?

简而言之,我正在为Maya构建插件。 核心插件是使用 C++ 构建的,但 GUI 是通过 Python 绘制的(因为它不是处理器密集型的)。 由于我目前还不知道如何从插件中return多个值,除了默认的MStatus ,因此要在 Python 中更新字典,我必须传递变量的名称,指向实现 GUI 的对象并且其中包含字典本身,添加到插件中,然后使用MGlobal::executePythonCommand()Maya的全局范围更新字典。

为此,我所做的是:

import time

class foo(bar):

    def __init__(self):
        super(foo, self).__init__()
        self.time = time.time() #almost guaranteed to be unique on a single computer

    def name(self):
        g = globals()
        for x in g:
            if isinstance(g[x], type(self)):
                if g[x].time == self.time:
                    return x
                    #or you could:
                    #return filter(None,[x if g[x].time == self.time else None for x in g if isinstance(g[x], type(self))])
                    #and return all keys pointing to object itself

我知道这不是globals许多键可能指向同一个对象的完美解决方案,例如:

a = foo()
b = a
b.name()
>>>b
or
>>>a

并且该方法不是线程安全的。 如果我错了,请纠正我。

至少这种方法通过获取指向对象本身的全局作用域中的任何变量的名称解决了我的问题,并将其作为参数传递给插件,供它在内部使用。

我在int (原始整数类)上尝试过这个,但问题是这些原始类没有被绕过(如果错误,请更正使用的技术术语)。 您可以重新实现int然后执行int = fooa = 3永远不会是foo的对象,而是原语的对象。 要克服这个问题,您必须使用a = foo(3)才能使a.name()工作。

在 python 2.7 和更新版本中,还有字典理解,这使它更短一些。 如果可能的话,我会使用 getattr 代替 eval (eval 是邪恶的),就像在最佳答案中一样。 Self 可以是任何具有您正在查看的上下文的对象。 它可以是一个对象或 locals=locals() 等。

{name: getattr(self, name) for name in ['some', 'vars', 'here]}

我正在研究类似的问题。 @S.Lott 说“如果你有变量列表,“发现”它们的名字有什么意义? 我的回答只是看看它是否可以完成,以及是否出于某种原因您想按类型将变量排序到列表中。 所以无论如何,在我的研究中,我遇到了这个线程,我的解决方案有点扩展并且基于@rlotun 解决方案。 另一件事,@unutbu 说,“这个想法有优点,但请注意,如果两个变量名引用相同的值(例如 True),那么可能会返回一个意外的变量名。” 在这个练习中,这是正确的,所以我通过对每种可能性使用类似于此的列表理解来处理它: isClass = [i for i in isClass if i != 'item'] 没有它,“项目”将出现在每个列表中。

__metaclass__ = type

from types import *

class Class_1: pass
class Class_2: pass
list_1 = [1, 2, 3]
list_2 = ['dog', 'cat', 'bird']
tuple_1 = ('one', 'two', 'three')
tuple_2 = (1000, 2000, 3000)
dict_1 = {'one': 1, 'two': 2, 'three': 3}
dict_2 = {'dog': 'collie', 'cat': 'calico', 'bird': 'robin'}
x = 23
y = 29
pie = 3.14159
eee = 2.71828
house = 'single story'
cabin = 'cozy'

isClass = []; isList = []; isTuple = []; isDict = []; isInt = []; isFloat = []; isString = []; other = []

mixedDataTypes = [Class_1, list_1, tuple_1, dict_1, x, pie, house, Class_2, list_2, tuple_2, dict_2, y, eee, cabin]

print '\nMIXED_DATA_TYPES total count:', len(mixedDataTypes)

for item in mixedDataTypes:
    try:
        # if isinstance(item, ClassType): # use this for old class types (before 3.0)
        if isinstance(item, type):
            for k, v in list(locals().iteritems()):
                if v is item:
                    mapping_as_str = k
                    isClass.append(mapping_as_str)
            isClass = [i for i in isClass if i != 'item']

        elif isinstance(item, ListType):
            for k, v in list(locals().iteritems()):
                if v is item:
                    mapping_as_str = k
                    isList.append(mapping_as_str)
            isList = [i for i in isList if i != 'item']

        elif isinstance(item, TupleType):
            for k, v in list(locals().iteritems()):
                if v is item:
                    mapping_as_str = k
                    isTuple.append(mapping_as_str)
            isTuple = [i for i in isTuple if i != 'item']

        elif isinstance(item, DictType):
            for k, v in list(locals().iteritems()):
                if v is item:
                    mapping_as_str = k
                    isDict.append(mapping_as_str)
            isDict = [i for i in isDict if i != 'item']

        elif isinstance(item, IntType):
            for k, v in list(locals().iteritems()):
                if v is item:
                    mapping_as_str = k
                    isInt.append(mapping_as_str)
            isInt = [i for i in isInt if i != 'item']

        elif isinstance(item, FloatType):
            for k, v in list(locals().iteritems()):
                if v is item:
                    mapping_as_str = k
                    isFloat.append(mapping_as_str)
            isFloat = [i for i in isFloat if i != 'item']

        elif isinstance(item, StringType):
            for k, v in list(locals().iteritems()):
                if v is item:
                    mapping_as_str = k
                    isString.append(mapping_as_str)
            isString = [i for i in isString if i != 'item']

        else:
            for k, v in list(locals().iteritems()):
                if v is item:
                    mapping_as_str = k
                    other.append(mapping_as_str)
            other = [i for i in other if i != 'item']

    except (TypeError, AttributeError), e:
        print e

print '\n isClass:', len(isClass), isClass
print '  isList:', len(isList), isList
print ' isTuple:', len(isTuple), isTuple
print '  isDict:', len(isDict), isDict
print '   isInt:', len(isInt), isInt
print ' isFloat:', len(isFloat), isFloat
print 'isString:', len(isString), isString
print '   other:', len(other), other

# my output and the output I wanted
'''
MIXED_DATA_TYPES total count: 14

 isClass: 2 ['Class_1', 'Class_2']
  isList: 2 ['list_1', 'list_2']
 isTuple: 2 ['tuple_1', 'tuple_2']
  isDict: 2 ['dict_1', 'dict_2']
   isInt: 2 ['x', 'y']
 isFloat: 2 ['pie', 'eee']
isString: 2 ['house', 'cabin']
   other: 0 []
'''

大多数对象没有__name__属性。 (类、函数和模块都可以;还有其他内置类型吗?)

除了print("my_var")之外,您对print(my_var.__name__)还有什么期望? 你可以直接使用字符串吗?

你可以“切片”一个字典:

def dict_slice(D, keys, default=None):
  return dict((k, D.get(k, default)) for k in keys)

print dict_slice(locals(), ["foo", "bar"])
# or use set literal syntax if you have a recent enough version:
print dict_slice(locals(), {"foo", "bar"})

或者:

throw = object()  # sentinel
def dict_slice(D, keys, default=throw):
  def get(k):
    v = D.get(k, throw)
    if v is not throw:
      return v
    if default is throw:
      raise KeyError(k)
    return default
  return dict((k, get(k)) for k in keys)

你可以使用easydict

>>> from easydict import EasyDict as edict
>>> d = edict({'foo':3, 'bar':{'x':1, 'y':2}})
>>> d.foo
3
>>> d.bar.x
1
>>> d = edict(foo=3)
>>> d.foo
3

另一个例子:

>>> d = EasyDict(log=False)
>>> d.debug = True
>>> d.items()
[('debug', True), ('log', False)]

在 python3 上,此函数将获取堆栈中最外层的名称:

import inspect


def retrieve_name(var):
        """
        Gets the name of var. Does it from the out most frame inner-wards.
        :param var: variable to get name from.
        :return: string
        """
        for fi in reversed(inspect.stack()):
            names = [var_name for var_name, var_val in fi.frame.f_locals.items() if var_val is var]
            if len(names) > 0:
                return names[0]

它在代码的任何地方都很有用。 遍历反向堆栈以查找第一个匹配项。

虽然这可能是一个糟糕的主意,但它与 rlotun 的答案大致相同,但它会更频繁地返回正确的结果。

import inspect
def getVarName(getvar):
  frame = inspect.currentframe()
  callerLocals = frame.f_back.f_locals
  for k, v in list(callerLocals.items()):
    if v is getvar():
      callerLocals.pop(k)
      try:
        getvar()
        callerLocals[k] = v
      except NameError:
        callerLocals[k] = v
        del frame
        return k
  del frame

你这样称呼它:

bar = True
foo = False
bean = False
fooName = getVarName(lambda: foo)
print(fooName) # prints "foo"

应该得到列表然后返回

def get_var_name(**kwargs):
    """get variable name
        get_var_name(var = var)
    Returns:
        [str] -- var name
    """
    return list(kwargs.keys())[0]

它不会返回变量的名称,但您可以轻松地从全局变量创建字典。

class CustomDict(dict):
    def __add__(self, other):
        return CustomDict({**self, **other})

class GlobalBase(type):
    def __getattr__(cls, key):
        return CustomDict({key: globals()[key]})

    def __getitem__(cls, keys):
        return CustomDict({key: globals()[key] for key in keys})

class G(metaclass=GlobalBase):
    pass

x, y, z = 0, 1, 2

print('method 1:', G['x', 'y', 'z']) # Outcome: method 1: {'x': 0, 'y': 1, 'z': 2}
print('method 2:', G.x + G.y + G.z) # Outcome: method 2: {'x': 0, 'y': 1, 'z': 2}

使用python-varname你可以轻松做到:

pip install python-varname

from varname import Wrapper

foo = Wrapper(True)
bar = Wrapper(False)

your_dict = {val.name: val.value for val in (foo, bar)}

print(your_dict)

# {'foo': True, 'bar': False}

免责声明:我是那个 python-varname 库的作者。

>>> a = 1
>>> b = 1
>>> id(a)
34120408
>>> id(b)
34120408
>>> a is b
True
>>> id(a) == id(b)
True

通过这种方式获取可能是“a”或“b”的 varname。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM