如何在Matlab中实现基于指定整数数组的floor类取整函数？

需求说明 #

• 实现类似floor的功能：将数值数组中每个元素，取为指定整数集中小于等于该元素的最大整数
• 后续可扩展实现类似ceil的功能：取指定整数集中大于等于该元素的最小整数
• 示例：
  数值数组 A = [-1.1778, 1.3657, 2.9880, -0.5674]
  指定整数集 Q = [-3, -1, 1, 3]
  向下取整结果应为 [-3, 1, 1, -3]

原代码问题分析 #

原代码通过计算每个元素与整数集的差值绝对值取最近邻元素，无法满足向下/向上取整的逻辑需求，需要调整判断逻辑。

向下取整函数实现（floorSet） #

function s_hat = floorSet(num_array, integer_set)
    % 先对整数集升序排序，确保查找逻辑的正确性
    sorted_Q = sort(integer_set);
    s_hat = zeros(size(num_array));
    
    for ii = 1:length(num_array)
        current_num = num_array(ii);
        % 筛选出整数集中小于等于当前数值的所有元素
        valid_elements = sorted_Q <= current_num;
        
        if any(valid_elements)
            % 取符合条件的最大元素（排序后最后一个有效索引对应的值）
            s_hat(ii) = sorted_Q(max(find(valid_elements)));
        else
            % 极端情况：当前数值小于整数集所有元素，取最小的元素
            s_hat(ii) = sorted_Q(1);
        end
    end
end

测试示例 #

A = [-1.1778, 1.3657, 2.9880, -0.5674];
Q = [-3, -1, 1, 3];
floor_result = floorSet(A, Q);
% 输出结果：[-3, 1, 1, -3]

向上取整函数实现（ceilingSet） #

基于相同逻辑，修改判断条件即可实现向上取整：

function s_hat = ceilingSet(num_array, integer_set)
    sorted_Q = sort(integer_set);
    s_hat = zeros(size(num_array));
    
    for ii = 1:length(num_array)
        current_num = num_array(ii);
        % 筛选出整数集中大于等于当前数值的所有元素
        valid_elements = sorted_Q >= current_num;
        
        if any(valid_elements)
            % 取符合条件的最小元素（排序后第一个有效索引对应的值）
            s_hat(ii) = sorted_Q(min(find(valid_elements)));
        else
            % 极端情况：当前数值大于整数集所有元素，取最大的元素
            s_hat(ii) = sorted_Q(end);
        end
    end
end

测试示例 #

A = [-1.1778, 1.3657, 2.9880, -0.5674];
Q = [-3, -1, 1, 3];
ceil_result = ceilingSet(A, Q);
% 输出结果：[-1, 3, 3, -1]

优化说明 #

• 先对整数集排序，避免因输入集无序导致的查找错误
• 直接通过逻辑筛选替代双重循环计算差值，逻辑更清晰，在大数据量场景下效率更高

内容的提问来源于stack exchange，提问作者Yash