如何在每个网格IE处按时间轴对气候数据集的特定纬度和经度上的时间轴进行排序

Question

我有一个气候数据集的NetCDF文件，其形状为20 * 445 * 445 AS（时间，纬度，经度），

我已将其读为 numpy.ma.core.maskedarray 使用 > netcdf4 库

array 在时间轴上的数据

我想对此 numpy 3D - （纬度，经度）该值应为上升或降序，

我尝试了

和

import netCDF4 as nc
from netCDF4 import Dataset,num2date,date2num
import sys,os,numpy as np,logging
from datetime import date, datetime, timedelta
import matplotlib.pyplot as plt

fileName = 'some_file'
data = Dataset(fileName, 'r')

temp_data = data.variables['Band1'][:,:,:]

数据集的原始数据集图

for i in range(0,20):
    data_to_plot = temp_data[i]
    
    fig, ax = plt.subplots(figsize = (4,4))
    ax.set_title('Layer '+ str(i))
    cax = ax.imshow(data_to_plot, cmap = plt.cm.Accent)
    
    cbar = plt.colorbar(cax,
                        orientation='vertical',
                        fraction=0.045, 
                        pad=0.05)
    ax.set_xlabel('X-axis')
    ax.set_ylabel('Y-axis')

sorted_temp_data = np.argsort(temp_data, axis=0)

排序

for i in range(0,20):
    data_to_plot = sorted_temp_data[i]
    
    fig, ax = plt.subplots(figsize = (4,4))
    ax.set_title('Layer '+ str(i))
    cax = ax.imshow(data_to_plot, cmap = plt.cm.Accent)
    
    cbar = plt.colorbar(cax,
                        orientation='vertical',
                        fraction=0.045, 
                        pad=0.05)
    
    ax.set_xlabel('X-axis')
    ax.set_ylabel('Y-axis')

如何做？

Answer 1

numpy的 argsort 函数不返回排序的数组。它返回排序的数组的索引。您可以直接沿时间轴（假设时间轴为0轴）直接对数组进行排序，如下所示：

np.sort(temp_data, axis=0)

如果要使用ArgSort，则代码应该是这样：

idx = np.argsort(temp_data, axis=0)
sort_data = np.take_along_axis(temp_data, idx,axis=0)