Criando uma estrutura de matriz tridimensional e escrevendo em arquivo mat em Python
Pergunta
Eu sou muito novo em Python.Preciso ter uma matriz tridimensional, para salvar a matriz 8 por 8 em algum comprimento.Vamos ligar para 530.O problema é que usei np.array, pois a matriz não pode ter mais de 2 dimensões, como argumenta numpy.
R = zeros([8,8,530],float)
Calculei minha matriz 8 por 8 como uma np.matrix
R[:,:,ii] = smallR
E então tento salvá-lo em um arquivo mat, como o scipy afirma fazer isso.
sio.savemat('R.mat',R)
No entanto, o erro diz que o objeto 'numpy.ndarray' não possui o atributo 'items'
/usr/local/lib/python2.7/dist-packages/scipy/io/matlab/mio.py:266: FutureWarning: Using oned_as default value ('column') This will change to 'row' in future versions
oned_as=oned_as)
Traceback (most recent call last):
File "ClassName.py", line 83, in <module>
print (buildR()[1])
File "ClassName.py", line 81, in buildR
sio.savemat('R.mat',R)
File "/usr/local/lib/python2.7/dist-packages/scipy/io/matlab/mio.py", line 269, in savemat
MW.put_variables(mdict)
File "/usr/local/lib/python2.7/dist-packages/scipy/io/matlab/mio5.py", line 827, in put_variables
for name, var in mdict.items():
AttributeError: 'numpy.ndarray' object has no attribute 'items'
Solução
Se você digitar help(sio.savemat)
, você vê:
savemat(file_name, mdict, appendmat=True, format='5', long_field_names=False, do_compression=False, oned_as=None)
Save a dictionary of names and arrays into a MATLAB-style .mat file.
[...]
mdict : dict
Dictionary from which to save matfile variables.
e mesmo que você não reconheça .items()
como método de dicionário, está claro que precisaremos usar um dicionário (um conjunto de pares de chave e valor;google "tutorial de dicionário python" se necessário).
Nesse caso:
>>> from numpy import zeros
>>> from scipy import io as sio
>>>
>>> R = zeros([8,8,530],float)
>>> R += 12.3
>>>
>>> sio.savemat('R.mat', {'R': R})
>>>
>>> S = sio.loadmat('R.mat')
>>> S
{'R': array([[[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3],
[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3],
[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3],
...,
...,
[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3],
[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3],
[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3]]]), '__version__': '1.0', '__header__': 'MATLAB 5.0 MAT-file Platform: posix, Created on: Sat Feb 25 18:16:02 2012', '__globals__': []}
>>> S['R']
array([[[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3],
[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3],
[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3],
...,
...,
[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3],
[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3],
[ 12.3, 12.3, 12.3, ..., 12.3, 12.3, 12.3]]])
Basicamente, um dicionário é usado para que os arrays possam ser nomeados, já que você pode armazenar vários objetos em um arquivo .mat.