使用 scipy.io.loadmat 将 .mat Matlab 文件中的字典键转换为 Python 中具有相同值的变量名

Question

我正在尝试使用 scipy.io.loadmat 从 .mat 文件加载基本字典 temp = {'key':array([1,2])} 。将 loadmat() 返回的 Python 字典文件中的键转换为变量名，其值与代表键相同。

例如：

temp = {'key':array([1,2])}

变成

key = array([1,2])

我知道如何用 temp.keys() 获取钥匙。然后抓取项目很容易，但如何强制 temp.keys() 中的字符串列表为变量名而不是字符串。

我希望这是有道理的，但这可能真的很容易，我只是不知道该怎么做。

干杯

Answer 1

在 python 中，方法参数可以使用 ** 魔法作为字典传递：

def my_func(key=None):
   print key
   #do the real stuff

temp = {'key':array([1,2])}

my_func(**temp)

>>> array([1,2])

Answer 2

最好的办法是使用 temp['key']。但是，要回答这个问题，您可以使用 exec 函数。这样做的好处是，您不必硬编码任何变量名或将自己限制在函数内工作。

from numpy import array,matrix

temp = {'key':array([1,2]),'b': 4.3,'c': 'foo','d':matrix([2,2])}

for k in temp:
    exec('{KEY} = {VALUE}'.format(KEY = k, VALUE = repr(temp[k])))

>>> key
array([1, 2])
>>> b
4.3
>>> c
'foo'
>>> d
matrix([[2, 2]])

注意：仅当您从模块导入特定功能时，此操作才有效。如果由于代码实践或需要导入的函数数量过多而不想这样做，则可以编写一个函数来在条目前面连接模块名称。输出与前面的示例相同。

import numpy as np,numpy

temp = {'key':np.array([1,2]),'b': 4.3,'c': 'foo','d':np.matrix([2,2])}

def exec_str(key,mydict):
    s = str(type(mydict[key]))
    if '.' in s:
        start = s.index("'") + 1
        end = s.index(".") + 1
        v = s[start:end:] + repr(mydict[key])
    else:
        v = repr(mydict[key])     
    return v

for k in temp:
    exec('{KEY} = {VALUE}'.format(KEY = k, VALUE = exec_str(k,temp)))

虽然这不是最佳的代码实践，但它对于我测试的所有示例都很有效。

Answer 3

更好的方法可能是将数据填充到一个单独的对象中：

class attrdict(dict):
    def __getattr__(self, k): return self[k]
    def __setattr__(self, k, v): self[k] = v

somedict = {'key': 123, 'stuff': 456}

data = attrdict(somedict)
print data.key
print data.stuff

这很容易交互使用，并且不需要任何魔法。
这对于 Matlab 用户来说也应该没问题。

编辑：事实证明，下面的内容在大多数情况下实际上不起作用。太糟糕了，魔法就这么多了。

不过，如果您想使用魔法，您可以执行类似这样的操作，

locals().update(somedict)

这将可以交互地正常工作，您甚至可以通过弄乱 sys 来隐藏对加载器函数内的 locals() 的访问._getframe().f_back.f_locals。

然而，这在函数中不起作用：

def foo():
    locals().update({'a': 4})
    print a 

重点是上面的 a 在编译时绑定到全局变量，因此 Python 不会尝试在局部变量中查找它。

Answer 4

正如 hdhagman 回答的那样，使用 exec() 更简单。我写了更简单的代码。

temp = {'key':[1,2]}

for k, v in temp.items():
    exec("%s = %s" % (k, v))

print(key)
=> [1,2]

Answer 5

对于 numpy 数组和其他非内置类型，上面的答案都不适合我。但是，执行了以下操作：

import numpy as np
temp = {'a':np.array([1,2]), 'b': 4.3, 'c': 'foo', 'd':np.matrix([2,2])}
for var in temp.keys():
    exec("{} = temp['{}']".format(var, var))

注意引号的顺序。这使得 var 在第一个实例中被视为变量，然后在第二个实例中被视为键，索引到 temp 字典中。

当然，关于 exec() 和 eval() 危险的常见免责声明仍然适用，并且您应该仅在您绝对信任的输入上运行此命令。

Answer 6

虽然我建议直接使用字典并访问诸如 temp['key'] 之类的数组，但如果您提前知道所有变量名称，您可以编写一个函数将它们提取到各个变量：

def func(**kwargs):
    return kwargs['a'],kwargs['b']


temp = {'a':np.array([1,2]),'b':np.array([3,4])}
a,b = func(**temp)
del temp # get rid of temporary dict

Answer 7

我有同样的问题，我根据其他人的答案尝试了这个代码。在这里，我使用 pickle 而不是 Matlab 来存储和检索数据。

保存数据

x1=[2,4,6,8,10]
x2=[1,3,5,7,9]
x3='hello world'
vars={'x1':'x1','x2':'x2','x3':'x3'}

file='test.pickle'
f = open(file,'wb')
for key in vars.keys():
    pickle.dump(eval(vars[key]),f)

del x1,x2,x3

从文件中读取

s = open(file,'rb')
for k in vars:
    exec('{KEY} = {VALUE}'.format(KEY = k, VALUE = repr(pickle.load(s))))

print(x1,x2,x3)