Earth Engine 支援陣列轉換,例如轉置、反向和偽反向。舉例來說,請考慮圖片時間序列的一般最小二乘法 (OLS) 迴歸。在以下範例中,我們將含有預測變數和回應頻帶的圖片轉換為陣列圖片,然後「求解」,以三種方式取得最小二乘係數估計值。首先,請組合圖片資料並轉換為陣列:
程式碼編輯器 (JavaScript)
// Scales and masks Landsat 8 surface reflectance images. function prepSrL8(image) { // Develop masks for unwanted pixels (fill, cloud, cloud shadow). var qaMask = image.select('QA_PIXEL').bitwiseAnd(parseInt('11111', 2)).eq(0); var saturationMask = image.select('QA_RADSAT').eq(0); // Apply the scaling factors to the appropriate bands. var opticalBands = image.select('SR_B.').multiply(0.0000275).add(-0.2); var thermalBands = image.select('ST_B.*').multiply(0.00341802).add(149.0); // Replace the original bands with the scaled ones and apply the masks. return image.addBands(opticalBands, null, true) .addBands(thermalBands, null, true) .updateMask(qaMask) .updateMask(saturationMask); } // Load a Landsat 8 surface reflectance image collection. var collection = ee.ImageCollection('LANDSAT/LC08/C02/T1_L2') // Filter to get only two years of data. .filterDate('2019-04-01', '2021-04-01') // Filter to get only imagery at a point of interest. .filterBounds(ee.Geometry.Point(-122.08709, 36.9732)) // Prepare images by mapping the prepSrL8 function over the collection. .map(prepSrL8) // Select NIR and red bands only. .select(['SR_B5', 'SR_B4']) // Sort the collection in chronological order. .sort('system:time_start', true); // This function computes the predictors and the response from the input. var makeVariables = function(image) { // Compute time of the image in fractional years relative to the Epoch. var year = ee.Image(image.date().difference(ee.Date('1970-01-01'), 'year')); // Compute the season in radians, one cycle per year. var season = year.multiply(2 * Math.PI); // Return an image of the predictors followed by the response. return image.select() .addBands(ee.Image(1)) // 0. constant .addBands(year.rename('t')) // 1. linear trend .addBands(season.sin().rename('sin')) // 2. seasonal .addBands(season.cos().rename('cos')) // 3. seasonal .addBands(image.normalizedDifference().rename('NDVI')) // 4. response .toFloat(); }; // Define the axes of variation in the collection array. var imageAxis = 0; var bandAxis = 1; // Convert the collection to an array. var array = collection.map(makeVariables).toArray(); // Check the length of the image axis (number of images). var arrayLength = array.arrayLength(imageAxis); // Update the mask to ensure that the number of images is greater than or // equal to the number of predictors (the linear model is solvable). array = array.updateMask(arrayLength.gt(4)); // Get slices of the array according to positions along the band axis. var predictors = array.arraySlice(bandAxis, 0, 4); var response = array.arraySlice(bandAxis, 4);
import ee import geemap.core as geemap
Colab (Python)
import math # Scales and masks Landsat 8 surface reflectance images. def prep_sr_l8(image): # Develop masks for unwanted pixels (fill, cloud, cloud shadow). qa_mask = image.select('QA_PIXEL').bitwiseAnd(int('11111', 2)).eq(0) saturation_mask = image.select('QA_RADSAT').eq(0) # Apply the scaling factors to the appropriate bands. optical_bands = image.select('SR_B.').multiply(0.0000275).add(-0.2) thermal_bands = image.select('ST_B.*').multiply(0.00341802).add(149.0) # Replace the original bands with the scaled ones and apply the masks. return ( image.addBands(optical_bands, None, True) .addBands(thermal_bands, None, True) .updateMask(qa_mask) .updateMask(saturation_mask) ) # Load a Landsat 8 surface reflectance image collection. collection = ( ee.ImageCollection('LANDSAT/LC08/C02/T1_L2') # Filter to get only two years of data. .filterDate('2019-04-01', '2021-04-01') # Filter to get only imagery at a point of interest. .filterBounds(ee.Geometry.Point(-122.08709, 36.9732)) # Prepare images by mapping the prep_sr_l8 function over the collection. .map(prep_sr_l8) # Select NIR and red bands only. .select(['SR_B5', 'SR_B4']) # Sort the collection in chronological order. .sort('system:time_start', True) ) # This function computes the predictors and the response from the input. def make_variables(image): # Compute time of the image in fractional years relative to the Epoch. year = ee.Image(image.date().difference(ee.Date('1970-01-01'), 'year')) # Compute the season in radians, one cycle per year. season = year.multiply(2 * math.pi) # Return an image of the predictors followed by the response. return ( image.select() .addBands(ee.Image(1)) # 0. constant .addBands(year.rename('t')) # 1. linear trend .addBands(season.sin().rename('sin')) # 2. seasonal .addBands(season.cos().rename('cos')) # 3. seasonal .addBands(image.normalizedDifference().rename('NDVI')) # 4. response .toFloat() ) # Define the axes of variation in the collection array. image_axis = 0 band_axis = 1 # Convert the collection to an array. array = collection.map(make_variables).toArray() # Check the length of the image axis (number of images). array_length = array.arrayLength(image_axis) # Update the mask to ensure that the number of images is greater than or # equal to the number of predictors (the linear model is solvable). array = array.updateMask(array_length.gt(4)) # Get slices of the array according to positions along the band axis. predictors = array.arraySlice(band_axis, 0, 4) response = array.arraySlice(band_axis, 4)
請注意,arraySlice()
會針對沿著 bandAxis
(1 軸) 指定的索引範圍,傳回時間序列中的所有圖片。此時,您可以使用矩陣代數學來解出 OLS 係數:
程式碼編輯器 (JavaScript)
// Compute coefficients the hard way. var coefficients1 = predictors.arrayTranspose().matrixMultiply(predictors) .matrixInverse().matrixMultiply(predictors.arrayTranspose()) .matrixMultiply(response);
import ee import geemap.core as geemap
Colab (Python)
# Compute coefficients the hard way. coefficients_1 = ( predictors.arrayTranspose() .matrixMultiply(predictors) .matrixInverse() .matrixMultiply(predictors.arrayTranspose()) .matrixMultiply(response) )
雖然這個方法可行,但效率不佳,而且會導致程式碼難以閱讀。更好的方法是使用 pseudoInverse()
方法 (matrixPseudoInverse()
適用於陣列圖片):
程式碼編輯器 (JavaScript)
// Compute coefficients the easy way. var coefficients2 = predictors.matrixPseudoInverse() .matrixMultiply(response);
import ee import geemap.core as geemap
Colab (Python)
# Compute coefficients the easy way. coefficients_2 = predictors.matrixPseudoInverse().matrixMultiply(response)
從可讀性和運算效率的角度來看,取得 OLS 係數的最佳方式是 solve()
(如果是陣列圖片,則為 matrixSolve()
)。solve()
函式會根據輸入內容的特性,決定如何最佳解決系統,並使用超定系統的偽逆矩陣、正方陣的逆矩陣,以及近似奇異矩陣的特殊技術:
程式碼編輯器 (JavaScript)
// Compute coefficients the easiest way. var coefficients3 = predictors.matrixSolve(response);
import ee import geemap.core as geemap
Colab (Python)
# Compute coefficients the easiest way. coefficients_3 = predictors.matrixSolve(response)
如要取得多頻帶圖像,請將陣列圖像投影至較低維度的空間,然後將其扁平化:
程式碼編輯器 (JavaScript)
// Turn the results into a multi-band image. var coefficientsImage = coefficients3 // Get rid of the extra dimensions. .arrayProject([0]) .arrayFlatten([ ['constant', 'trend', 'sin', 'cos'] ]);
import ee import geemap.core as geemap
Colab (Python)
# Turn the results into a multi-band image. coefficients_image = ( coefficients_3 # Get rid of the extra dimensions. .arrayProject([0]).arrayFlatten([['constant', 'trend', 'sin', 'cos']]) )
請檢查這三種方法的輸出內容,並觀察無論使用哪種解算器,係數的最終矩陣都相同。solve()
靈活且高效,因此是一般用途線性建模的絕佳選擇。