matlab：对于一维数组，取出k个最靠近0的正数和负数以及对应的索引

原创已于 2022-11-18 11:42:15 修改 · 1.2k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#matlab #最靠近0的正数和负数

于 2022-11-18 11:39:37 首次发布

matlab编程专栏收录该内容

29 篇文章

订阅专栏

本文介绍了一种MATLAB函数Near0，该函数用于找出数组中距离0最近的指定数量的正数和负数及其索引。通过示例演示了如何使用此函数，并详细解释了其内部实现过程。

1. 示例

1.1 程序

Input = [-1,1,2,3,-2];
[Positive_index,Negative_index] = Near0(Input,2)  % 得到最靠近0的2个正数对应的索引以及负数对应的索引
Positive_num = Input(Positive_index)              % 最靠近0的2个正数
Negative_num = Input(Negative_index)              % 最靠近0的2个负数

1.2 输出

在这里插入图片描述

2. 函数

function [Positive_index,Negative_index] = Near0(Input,k)

% 如果只输入数组，默认k=1
if nargin == 1
    k = 1;
end

% 可以一行多列输入，也可以多行一列输入
if size(Input,1) == 1
    Input = Input';
end

% 最靠近0的负索引
index = find(Input<0);
data = Input(index,:);
[~,b] = sort(data,"descend");
Negative_index = index(b(1:k));

% 最靠近0的正索引
index = find(Input>0);
data = Input(index,:);
[~,b] = sort(data,"ascend");
Positive_index = index(b(1:k));