如何找到一个长度相同的多个数组的最小值的数组?

问题描述 投票:-1回答:1

我有一个多维网格数组,尺寸为(29,320,180),其中29是数组的数量,320是纬度值,180是经度值。我想从29个数组中找到每个网格点的最小值,所以最后我可以得到一个尺寸为320x180的数组,由每个网格点的最小值组成。我必须破坏每个数组都有大量的nan值。例如两个尺寸相同的数组:a=[[1,2,3],[3,5,8],[4,8,12]]b=[[3,5,6],[9,12,5],[5,6,14]],想要输出的是一个在每个索引处都有最小值的数组,也就是说:c=[[1,2,3],[3,5,5],[4,6,12]]。

python arrays min
1个回答
0
投票

我不知道你需要的是每个数组的列数还是行数的最小值,你可以通过下面的例子选择你想要的最小值。

让我们来创建一个几个小型二维数组的例子。

import numpy as np
ex_dict = {}
lat_min = []
lon_min = []
# creating fake data assuming instead of the 29 arrays of dimensions 320x180 you have 5 arrays of dimensions 2x5 (so we can see the output) and all the arrays are stored in a dictionnary (because it's easier for me to randomly create them that way :)
for i in range(0,5):
    ex_dict[i] = np.stack([np.random.choice(range(i,20), 5, replace=False) for _ in range(2)])

让我们来看看我们的数组。

ex_dict
{0: array([[19, 18,  5, 13,  6],
        [ 5, 12,  3,  8,  0]]),
 1: array([[10, 13,  2, 19, 15],
        [ 5, 19,  6,  8, 14]]),
 2: array([[ 5, 17, 10, 11,  7],
        [19,  2, 11,  5,  6]]),
 3: array([[14,  3, 17,  4, 11],
        [18, 10,  8,  3,  7]]),
 4: array([[15,  8, 18, 14, 10],
        [ 5, 19, 12, 16, 13]])}

然后我们创建一个列表来存储每个数组的最小值 (lat_min包含每个原始数组的最小值,lat_lon包含所有数组中每个列的最小值):

# for each of the 5 arrays (in this example, stored in the ex_dict dictionnary), find the minimum in each row (axis = 1) and each column (axis = 2)
for i in ex_dict:
    lat_min.append(np.nanmin(ex_dict[i], axis=1))
    lon_min.append(np.nanmin(ex_dict[i], axis=0)) 

我们的最小值列表:

lat_min
[array([5, 0]), array([2, 5]), array([5, 2]), array([3, 3]), array([8, 5])]

lon_min
[array([ 5, 12,  3,  8,  0]),
 array([ 5, 13,  2,  8, 14]),
 array([ 5,  2, 10,  5,  6]),
 array([14,  3,  8,  3,  7]),
 array([ 5,  8, 12, 14, 10])]
© www.soinside.com 2019 - 2024. All rights reserved.