多くのファイルを1つの配列に読み込む - Python

Question

それで、プロットするためにいくつかの機能を持つ多くの.matファイルを読み込む必要があります。

プロットされる各配列が辞書にロードされます。

import numpy as np 
import scipy.io as io 


dict1 = io.loadmat('file1.MAT') 
dict2 = io.loadmat('file2.MAT') # type = dict 
dict3 = io.loadmat('file3.MAT') 
... 

ので、私は後にプロットするために、私は必要dictionarieの要素を取らなければならない。この後

array1 = dict1['data'] 
array2 = dict2['data'] 
array3 = dict3['data'] 
... 

、私はプロットすることができますデータ。それは動作しますが、私には馬鹿に見えます（もし私が100のベクトルを持っていれば、時間がかかるでしょう...）。このタスクを実行するためのより良い方法はありますか？

Answer 1

多くの行列を扱うことについて話しているので、それらをコレクションとして管理する必要があります。まず、ファイルのセットを定義しましょう。彼らはパターンを持っているために起こる場合は、あなたが名前を生成してみてください

Matrix_files = [ 'fileA.MAT', 'file1.MAT', 'no pattern to these names.MAT' ] 

：それはタプル、またはリストすることができ

Matrix_files = [ 'file{}.MAT'.format(num) for num in range(1,4) ] 

彼らは共通の場所を共有する場合は、使用して検討するかもしれませんさまざまなディレクトリスキャンアプローチの1つ（opendirまたはglob、名前2）。

あなたはファイル名のリストを持っていたらあなたが辞書を読むことができ、：それと

def read_matrix(filespec): 
    from scipy.io import loadmat 
    md = loadmat(filespec) 
    # process md 
    return md 

、あなたはすべてのデータを取得し、またはデータの一部を取得することができ、次のいずれか

All_data = [read_matrix(f) for f in Matrix_files] 

Some_data = [read_matrix(f)['data'] for f in Matrix_files] 

データのみを気にする場合は、関数定義をスキップすることができます。

from scipy.io import loadmat 
Just_data = [loadmat(f)['data'] for f in Matrix_files]