matlab最大的未知维度的数组

时间:2013-01-03 12:53:11

标签: arrays matlab matrix max

我想计算最大值,更重要的是,计算N-by-N ... by-N数组的坐标,而不指定其尺寸。

例如,我们采取:

A = [2 3];
B = [2 3; 3 4];

该函数(让我们称之为MAXI)应该为矩阵A返回以下值:

[fmax, coor] = MAXI(A)

fmax =
   3

coor =
   2

和矩阵B

[fmax, coor] = MAXI(B)

fmax =
    4

coor=
    2   2

主要问题不是开发一个特别适用于某个类的代码,而是开发一个尽可能快地适用于任何输入(具有更高维度)的代码。

1 个答案:

答案 0 :(得分:8)

要查找绝对最大值,您必须先将输入矩阵转换为列向量,然后找到最大元素的线性索引,然后将其转换为ind2sub的坐标。这可能有点棘手,因为ind2sub需要指定已知数量的输出变量。为此,我们可以使用单元格数组和comma-separated lists,如下所示:

[fmax, coor] = max(A(:));
if ismatrix(A)
    C = cell(1:ndims(A));
    [C{:}] = ind2sub(size(A), coor);
    coor = cell2mat(C);
end

编辑:我添加了一个额外的if语句,用于检查输入是矩阵还是向量,如果是后者,则按原样返回线性索引。

在一个函数中,它看起来像这样:

function [fmax, coor] = maxi(x)
    [fmax, coor] = max(A(:));
    if ismatrix(A)
        C = cell(1:ndims(A));
        [C{:}] = ind2sub(size(A), coor);
        coor = cell2mat(C);
    end

实施例

A = [2 3; 3 4];
[fmax, coor] = maxi(A)

fmax =
    4

coor =
    2    2