コード例 #1
0
ファイル: LoadDataYear.py プロジェクト: ecoffel/climate-som
def loadDataYear(year, model, var , latindex = -1, lonindex = -1, reallat = -1, reallon = -1):     
    data1d = []
    lat=[]
    lon=[]
    
    baseDir = 'data/' + model + '/' + var + '/'
    
    for i in range(1,13):
        t = ""
        if i < 10:
            t = var + '_' + str(year) + '_0' + str(i) + '_01'
        else:
            t = var + '_' + str(year) + '_' + str(i) + '_01'
        data = scipy.io.loadmat(baseDir + t + '.mat')
        data = data[t][0]
        
        if latindex == -1 and lonindex == -1 and (reallat != -1 and reallon != -1):
            latindex = LatLonIndex.findLatIndex(reallat, data)
            latindex = latindex[0]
            lonindex = LatLonIndex.findLonIndex(reallon, data)
            lonindex = lonindex[1]
        
        if len(lat)==0:
            lat=data[0]
            lon=data[1]
            
        for d in range(0, len(data[2][latindex,lonindex])):
            data1d.append(data[2][latindex,lonindex,d])

    return [lat[latindex,lonindex],lon[latindex,lonindex],data1d]
コード例 #2
0
def loadDataYear(year,
                 model,
                 var,
                 latindex=-1,
                 lonindex=-1,
                 reallat=-1,
                 reallon=-1):
    data1d = []
    lat = []
    lon = []

    baseDir = 'data/' + model + '/' + var + '/'

    for i in range(1, 13):
        t = ""
        if i < 10:
            t = var + '_' + str(year) + '_0' + str(i) + '_01'
        else:
            t = var + '_' + str(year) + '_' + str(i) + '_01'
        data = scipy.io.loadmat(baseDir + t + '.mat')
        data = data[t][0]

        if latindex == -1 and lonindex == -1 and (reallat != -1
                                                  and reallon != -1):
            latindex = LatLonIndex.findLatIndex(reallat, data)
            latindex = latindex[0]
            lonindex = LatLonIndex.findLonIndex(reallon, data)
            lonindex = lonindex[1]

        if len(lat) == 0:
            lat = data[0]
            lon = data[1]

        for d in range(0, len(data[2][latindex, lonindex])):
            data1d.append(data[2][latindex, lonindex, d])

    return [lat[latindex, lonindex], lon[latindex, lonindex], data1d]