Question

在这里发布其他帖子后，我有一个函数根据其名称输出有关变量的信息。我想把它移到一个模块中。

#python 2.7
import numpy as np
def oshape(name):
    #output the name, type and shape/length of the input variable(s)
    #for array or list
    x=globals()[name]
    if type(x) is np.array or type(x) is np.ndarray:
        print('{:20} {:25} {}'.format(name, repr(type(x)), x.shape))
    elif type(x) is list:
        print('{:20} {:25} {}'.format(name, repr(type(x)), len(x)))
    else:
        print('{:20} {:25} X'.format(name, type(t)))

a=np.array([1,2,3])
b=[4,5,6]
oshape('a')
oshape('b')

输出：

a                    <type 'numpy.ndarray'>    (3,)
b                    <type 'list'>             3

我想将这个函数oshape（）放入一个模块中，以便可以重用它。但是，放置在模块中不允许从主模块访问全局变量。我尝试过'import __main__'之类的东西，甚至存储函数globals（）并将其传递给子模块。问题是globals（）是一个函数，它专门返回调用它的模块的全局变量，而不是每个模块的不同函数。

import numpy as np
import olib

a=np.array([1,2,3])
b=[4,5,6]

olib.oshape('a')
olib.oshape('b')

给我：

KeyError: 'a'

额外信息：目标是减少冗余类型。稍作修改（我把它拿出去使问题变得更简单），oshape可以报告一个变量列表，所以我可以使用它：

oshape('a', 'b', 'other_variables_i_care_about')

所以需要两次输入变量名称的解决方案并不是我想要的。此外，只是传入变量不允许打印名称。考虑在长日志文件中使用它来显示计算结果＆amp;检查变量大小。

Answer 1

您遇到的实际问题是命名空间问题。

你可以用这种方式编写你的方法：

def oshape(name, x):
    # output the name, type and shape/length of the input variable(s)
    # for array or list
    if type(x) in (np.array, np.ndarray):
        print('{:20} {:25} {}'.format(name, repr(type(x)), x.shape))
    elif type(x) is list:
        print('{:20} {:25} {}'.format(name, repr(type(x)), len(x)))
    else:
        print('{:20} {:25} X'.format(name, type(x)))

并像这样使用它：

import numpy as np
import olib

a=np.array([1,2,3])
b=[4,5,6]

olib.oshape('a', a)
olib.oshape('b', b)

但是在参数中包含变量及其名称看起来非常多余。

另一个解决方案是将globals() dict赋予方法并保留代码。

通过模块查看关于全局变量可见性的this answer。

Answer 2

您的逻辑过于复杂，您应该自己传递数组，因为您还将变量名称作为字符串传递，因此您不会查找您无权访问的内容。但是如果你想让你的代码完全正常工作，你可以在模块上设置一个属性：

import numpy as np
import olib

a = np.array([1, 2, 3])
b = [4, 5, 6]
olib.a = a
olib.b = b
olib.oshape('a')
olib.oshape('b')

这将获取任何参数并搜索模块，以便为attrs运行代码：

import numpy as np
import sys
from os.path import basename
import imp

def oshape(*args):
    # output the name, type and shape/length of the input variable(s)
    # for array or list
    file_name = sys.argv[0]
    mod = basename(file_name).split(".")[0]
    if mod  not in sys.modules:
        mod = imp.load_source(mod, file_name)
        for name in args:
            x = getattr(mod, name)
            if type(x) is np.array or type(x) is np.ndarray:
                print('{:20} {:25} {}'.format(name, repr(type(x)), x.shape))
            elif type(x) is list:
                print('{:20} {:25} {}'.format(name, repr(type(x)), len(x)))
            else:
                print('{} {} X'.format(name, type(x)))

只需传递变量名称字符串：

:~/$ cat t2.py 
import numpy as np
from olib import oshape

a = np.array([1, 2, 3])
b = [4, 5, 6]
c = "a str"

oshape("a", "b", "c")


:$ python t2.py 
a                    <type 'numpy.ndarray'>    (3,)
b                    <type 'list'>             3
c <type 'str'> X

在模块

2 个答案: