Matlab中含NaN的m×n矩阵列非NaN值的笛卡尔积构造

嘿，这个需求我熟！咱们一步步来实现你要的矩阵B，完美匹配核心逻辑～

核心思路 #

我们需要先从矩阵A的每一列提取非NaN值，生成这些集合的笛卡尔积，最后把每个积的元组按行排列成矩阵B。下面是具体实现步骤和代码：

步骤1：提取每一列的非NaN值 #

先遍历矩阵A的每一列，把每一列里的非NaN值单独提取出来，存储成cell数组（因为每列的有效元素个数可能不一样）：

% 定义示例矩阵A
A = [2.1 0 NaN; 69 NaN 1; NaN 32.1 NaN];

% 提取每一列的非NaN值，返回cell数组colVals
colVals = arrayfun(@(c) A(~isnan(A(:,c)), c), 1:size(A,2), 'UniformOutput', false);

对于你的示例，colVals会是包含3个元素的cell数组，分别对应每列的有效数值：[2.1; 69]、[0; 32.1]、[1]。

步骤2：生成笛卡尔积网格 #

用ndgrid工具生成所有可能的笛卡尔积组合，它会为每个维度生成对应网格矩阵：

% 生成笛卡尔积的网格变量，存储在cell数组gridVars中
[gridVars{1:size(A,2)}] = ndgrid(colVals{:});

步骤3：转换为目标矩阵B #

把每个网格变量转成列向量，再拼接起来，就得到每行对应一个笛卡尔积元组的矩阵B：

% 将每个网格变量转为列向量，再拼接成矩阵B
B = cell2mat(cellfun(@(x) x(:), gridVars, 'UniformOutput', false));

完整代码&测试 #

整合所有步骤，加上异常处理（避免全NaN列导致报错）：

% 定义输入矩阵A
A = [2.1 0 NaN; 69 NaN 1; NaN 32.1 NaN];

% 步骤1：提取每列非NaN值
colVals = arrayfun(@(c) A(~isnan(A(:,c)), c), 1:size(A,2), 'UniformOutput', false);

% 可选：检查是否有全NaN的列，避免后续出错
emptyCols = cellfun(@isempty, colVals);
if any(emptyCols)
    warning('存在全为NaN的列，无法生成有效笛卡尔积');
    B = [];
    return;
end

% 步骤2：生成笛卡尔积网格
[gridVars{1:size(A,2)}] = ndgrid(colVals{:});

% 步骤3：构造矩阵B
B = cell2mat(cellfun(@(x) x(:), gridVars, 'UniformOutput', false));

运行后得到的B是：

B =
    2.1000    0.0000    1.0000
    69.0000    0.0000    1.0000
    2.1000   32.1000    1.0000
    69.0000   32.1000    1.0000

这是严格基于列非NaN值的笛卡尔积结果（2×2×1=4个组合）。如果你的示例是想处理行的非NaN值，只需要先把A转置，运行代码后再把B转置回来即可。

额外说明 #

• 这个方法支持任意m×n的实值矩阵，不管每列的非NaN值个数差异多大
• 若某一列全为NaN，代码会给出警告并返回空矩阵，你可以根据需求调整异常处理逻辑

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