从形状数组(10,21,2)中获取三维最大值,得到最终数组(10,21)

2024-04-20 09:07:04 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个状态值矩阵-

in[]: q

out[]: array([[[  1.78571429e-01,   4.00000000e-01],
    [  2.92307692e-01,   3.91304348e-01],
    [  3.93939394e-01,   4.21052632e-01],
    [  4.41176471e-01,   2.83916084e-01],
    [  1.48148148e-01,  -8.08080808e-02],
    [  4.08450704e-01,   2.94117647e-01],
    [  1.87500000e-01,   4.34782609e-02],
    [  3.05555556e-01,   4.60000000e-01],
    [  3.75000000e-01,  -6.66666667e-02],
    [  2.23880597e-01,   5.00000000e-01],
    [  2.41379310e-01,   5.00000000e-01],
    [  5.00000000e-01,  -6.66666667e-02],
    [  8.33333333e-02,   3.68421053e-01],
    [  5.00000000e-01,   1.80555556e-01],
    [  4.00000000e-01,   3.84615385e-01],
    [  2.94117647e-01,   2.13615023e-01],
    [  4.60000000e-01,   1.25000000e-01],
    [  7.46031746e-01,   3.58024691e-01],
    [  1.00000000e+00,   1.59420290e-01],
    [  1.00000000e+00,  -6.66666667e-02],
    [  0.00000000e+00,   0.00000000e+00]],

   [[  4.68750000e-01,   1.48642430e-01],
    [  3.33333333e-02,   2.81004710e-01],
    [  6.25000000e-02,   2.09302326e-01],
    [  3.44262295e-01,   2.83582090e-01],
    [  3.60000000e-01,  -1.13235294e-01],
    [  2.94117647e-01,   3.51351351e-01],
    [  2.54237288e-01,   1.25000000e-01],
    [  1.90476190e-01,   7.50000000e-01],
    [  2.83018868e-01,   4.54545455e-01],
    [  2.09302326e-01,   4.25499232e-01],
    [  3.33333333e-01,   1.25000000e-01],
    [  4.16666667e-01,   3.66666667e-01],
    [  2.00000000e-01,   4.25499232e-01],
    [ -1.00000000e-01,   6.66666667e-02],
    [  5.23809524e-01,   6.00000000e-01],
    [  1.25000000e-01,  -6.97478992e-01],
    [  1.44444444e-01,   1.44444444e-01],
    [ -3.33333333e-01,   7.50000000e-01],
    [  1.00000000e+00,  -1.00000000e+00],
    [  1.00000000e+00,   4.54545455e-01],
    [  0.00000000e+00,   0.00000000e+00]],

   [[  2.28070175e-01,   3.23076923e-01],
    [  3.23076923e-01,   2.61538462e-01],
    [  3.75000000e-01,   2.09176788e-01],
    [  6.12244898e-02,   3.33333333e-01],
    [  2.32876712e-01,   2.25464191e-01],
    [  2.42424242e-01,   8.04597701e-02],
    [  2.64705882e-01,   1.91394777e-01],
    [  2.25806452e-01,   2.09176788e-01],
    [  4.76923077e-01,   5.55555556e-02],
    [  8.88888889e-02,   1.42857143e-01],
    [  2.41379310e-01,   2.06171108e-01],
    [  3.33333333e-01,   1.00000000e+00],
    [  2.30769231e-01,   2.00000000e-01],
    [  2.72727273e-01,   4.08163265e-02],
    [  2.00000000e-01,   2.22079589e-01],
    [ -6.66666667e-02,   2.22079589e-01],
    [  5.55555556e-02,  -1.00000000e+00],
    [  6.66666667e-01,   4.08163265e-02],
    [  6.36363636e-01,  -9.09090909e-02],
    [  1.00000000e+00,   2.14285714e-01],
    [  0.00000000e+00,   0.00000000e+00]],

   [[  2.18750000e-01,   7.14285714e-02],
    [  5.26315789e-02,   2.13921902e-01],
    [  2.00000000e-01,   2.08888889e-01],
    [ -1.61290323e-01,   9.72222222e-01],
    [ -1.01449275e-01,   1.70568562e-01],
    [  1.27272727e-01,   1.04370891e-01],
    [  2.46753247e-01,   8.33333333e-02],
    [  2.35294118e-01,   1.66666667e-01],
    [  2.77777778e-02,   2.30769231e-01],
    [ -9.67741935e-02,   2.30769231e-01],
    [  3.57142857e-01,  -1.52941176e-01],
    [  2.38095238e-01,  -9.09090909e-02],
    [  2.63157895e-01,   1.81818182e-01],
    [  8.33333333e-02,   1.00000000e+00],
    [ -4.76190476e-02,  -1.11111111e-01],
    [  2.30769231e-01,   1.81818182e-01],
    [ -1.52941176e-01,   1.30434783e-01],
    [  6.92307692e-01,   2.50000000e-01],
    [  6.66666667e-01,  -1.11111111e-01],
    [  1.00000000e+00,   1.31313131e-01],
    [  0.00000000e+00,   0.00000000e+00]],

   [[  1.60000000e-01,   1.64179104e-01],
    [ -3.22580645e-02,   1.49253731e-02],
    [  4.11764706e-01,   2.54237288e-01],
    [ -1.44927536e-02,   1.30434783e-01],
    [  1.00000000e-01,  -5.08474576e-02],
    [  1.50684932e-01,  -1.42857143e-01],
    [ -7.93650794e-02,   1.50684932e-01],
    [  2.66666667e-01,   1.66666667e-01],
    [  1.66666667e-01,  -4.54545455e-02],
    [  1.42857143e-01,  -4.54545455e-02],
    [  1.30434783e-01,   1.81818182e-01],
    [  5.88235294e-02,   1.66666667e-01],
    [ -9.09090909e-02,   1.53846154e-01],
    [  2.50000000e-01,  -1.42857143e-01],
    [ -7.69230769e-02,   1.50684932e-01],
    [  1.76470588e-01,  -2.65432099e-01],
    [  2.50000000e-01,  -4.54545455e-02],
    [  1.37500000e-01,   2.41379310e-01],
    [  1.00000000e+00,   1.81818182e-01],
    [  6.00000000e-01,   1.81818182e-01],
    [  0.00000000e+00,   0.00000000e+00]],

   [[ -6.06060606e-02,  -1.85185185e-01],
    [  1.66666667e-01,  -5.26315789e-02],
    [  1.86440678e-01,   3.44827586e-02],
    [ -1.61290323e-01,   3.91304348e-01],
    [  3.84615385e-02,   2.00000000e-01],
    [ -9.80392157e-02,   1.80645161e-01],
    [  1.11111111e-01,  -1.61290323e-01],
    [  1.52542373e-01,   8.18181818e-01],
    [  2.94117647e-02,   5.29411765e-01],
    [  1.03448276e-01,  -9.80392157e-02],
    [  3.91304348e-01,   8.00000000e-01],
    [ -2.77555756e-17,  -1.00000000e+00],
    [  3.44827586e-02,  -1.05820106e-01],
    [  2.00000000e-01,  -4.34782609e-02],
    [  5.29411765e-01,   8.18181818e-01],
    [  2.72727273e-01,   4.97737557e-02],
    [  1.11111111e-01,  -1.86440678e-01],
    [  6.14035088e-01,  -1.05820106e-01],
    [  8.18181818e-01,  -3.70370370e-02],
    [  1.00000000e+00,  -1.00000000e+00],
    [  0.00000000e+00,   0.00000000e+00]],

   [[ -1.22807018e-01,   7.24637681e-02],
    [ -7.93650794e-02,   1.21212121e-01],
    [ -7.69230769e-02,   6.66666667e-01],
    [  1.34328358e-01,   2.05479452e-01],
    [  7.24637681e-02,   1.00000000e-01],
    [ -1.42857143e-01,  -2.14285714e-01],
    [  2.28070175e-01,   4.76190476e-02],
    [  4.76190476e-02,   5.56708673e-01],
    [  1.66666667e-01,  -2.14285714e-01],
    [ -2.11267606e-01,   5.05912620e-01],
    [  1.00000000e-01,   5.68627451e-01],
    [  2.50000000e-01,   1.00000000e+00],
    [ -3.33333333e-01,   5.56708673e-01],
    [ -3.04347826e-01,   5.64705882e-01],
    [  0.00000000e+00,  -5.15570934e-01],
    [  1.66666667e-01,   0.00000000e+00],
    [  5.05912620e-01,   5.68627451e-01],
    [  5.78947368e-01,   5.05912620e-01],
    [  1.00000000e+00,   7.86599022e-02],
    [  1.00000000e+00,  -5.42483660e-01],
    [  0.00000000e+00,   0.00000000e+00]],

   [[ -1.91489362e-01,  -2.90322581e-01],
    [ -2.04081633e-02,  -1.32231405e-01],
    [ -4.10958904e-02,  -6.56790123e-02],
    [ -2.92307692e-01,  -5.67733990e-01],
    [ -3.12500000e-02,  -2.94117647e-01],
    [ -2.90322581e-01,  -2.25806452e-01],
    [ -4.54545455e-01,  -5.26315789e-02],
    [ -1.29032258e-01,  -1.52542373e-01],
    [ -7.50000000e-02,  -9.23629929e-02],
    [ -7.46268657e-02,  -2.91666667e-01],
    [ -2.25806452e-01,  -1.52542373e-01],
    [ -2.94117647e-01,  -2.91666667e-01],
    [  5.88235294e-02,  -1.97090909e-01],
    [  5.88235294e-02,  -5.67733990e-01],
    [ -1.42857143e-01,  -1.63636364e-01],
    [ -5.26315789e-02,  -4.73170732e-01],
    [ -2.91666667e-01,  -1.57894737e-01],
    [ -2.50000000e-01,   8.18181818e-01],
    [  1.00000000e+00,  -5.34482759e-01],
    [  1.00000000e+00,  -1.00000000e+00],
    [  0.00000000e+00,   0.00000000e+00]],

   [[  8.77192982e-02,  -3.63636364e-01],
    [  1.42857143e-01,  -5.71428571e-02],
    [ -5.71428571e-02,  -1.72167707e-01],
    [ -3.33333333e-01,  -7.24637681e-02],
    [ -3.23076923e-01,  -2.00595238e-01],
    [ -2.18750000e-01,  -8.33333333e-02],
    [ -2.50000000e-01,   1.11111111e-01],
    [ -3.51351351e-01,  -7.24637681e-02],
    [ -2.20338983e-01,  -2.00595238e-01],
    [ -3.23529412e-01,   1.06227106e-01],
    [ -1.20000000e-01,  -1.87500000e-01],
    [ -5.20000000e-01,  -1.42857143e-01],
    [ -4.54545455e-01,  -2.00595238e-01],
    [ -1.42857143e-01,   1.06227106e-01],
    [ -1.42857143e-01,  -2.00595238e-01],
    [  1.11111111e-01,   1.98412698e-01],
    [ -1.17647059e-01,   1.06227106e-01],
    [  1.06227106e-01,  -4.00000000e-01],
    [  8.57142857e-01,  -1.87500000e-01],
    [  6.92307692e-01,  -1.11111111e-01],
    [  0.00000000e+00,   0.00000000e+00]],

   [[ -6.12244898e-02,  -3.06122449e-01],
    [ -3.33333333e-01,  -2.72727273e-01],
    [ -2.78688525e-01,  -3.23936961e-01],
    [ -9.43396226e-02,  -3.20000000e-01],
    [  6.06060606e-02,  -2.80000000e-01],
    [ -2.69841270e-01,  -2.38095238e-01],
    [ -3.20000000e-01,  -3.20000000e-01],
    [ -2.00000000e-01,   1.40000000e-01],
    [ -3.82352941e-01,   2.26190476e-01],
    [ -2.63157895e-01,  -2.98245614e-01],
    [ -2.72727273e-01,   1.40000000e-01],
    [ -3.68421053e-01,   1.40000000e-01],
    [ -3.00000000e-01,  -1.09230324e-02],
    [ -3.75000000e-01,   1.40000000e-01],
    [ -2.00000000e-01,   2.77555756e-17],
    [  2.00000000e-01,  -5.86428571e-01],
    [  2.95138889e-02,  -3.26923077e-01],
    [  2.26190476e-01,   2.26190476e-01],
    [  5.00000000e-02,   3.33333333e-01],
    [  5.00000000e-01,  -2.18750000e-01],
    [  0.00000000e+00,   0.00000000e+00]]])

我想从q[:,:,0]和q[:,:,1]中找到对应值的最大值,从而得到形状(10,21)的矩阵


Tags: in矩阵outarray形状状态值
1条回答
网友
1楼 · 发布于 2024-04-20 09:07:04

您可以使用numpy.amax,它返回数组的最大值或沿轴的最大值。在

示例(您的案例):

在:np.amax(np.array([[[1,2],[5,6]],[[10,11],[15,16]]]), 2)

输出: array([[ 2, 6], [11, 16]])

相关问题 更多 >