JavaScript对象属性的笛卡尔积

8

我有一个以下形式的对象(下面是简化的测试案例)

var test = {
        shirts: {
            sizes: ['large', 'medium']
            ,colors:['red', 'blue']
        }
        , trousers: {
            type: ['formal', 'casual']
            , pattern: ['plaid', 'stripes']
        }
    };

我希望生成属性的笛卡尔积,使输出为以下形式的数组:
// desired output

[ {shirts:{sizes:'large', color:'red'}, trousers:{type:'formal', pattern:'plaid'}}
  ,{shirts:{sizes:'large', color:'red'}, trousers:{type:'formal', pattern:'stripes'}}
  ,{shirts:{sizes:'large', color:'red'}, trousers:{type:'casual', pattern:'plaid'}}
  , {shirts:{sizes:'large', color:'red'}, trousers:{type:'casual', pattern:'stripes'}}
  ,{shirts:{sizes:'large', color:'blue'}, trousers:{type:'formal', pattern:'plaid'}}
..... and so on  ]

我该如何实现这一点?我根据另一个SO帖子上的数组笛卡尔积修改了以下代码,但似乎陷入了纠结之中,试图让其正常运行。
 function myCartesianProduct(input, current) {
    if (!input) { return []; }


    var head = input[Object.keys(input)[0]];

    var tail = objSlice(input);

    var output = [];


    for (var key in head) {

        for (var i = 0; i < head[key].length; i++) {

            var newCurrent = copy(current);

            newCurrent[key] = head[key][i];


            if (Object.keys(tail).length) {   //if tail.length
                var productOfTail =
                        myCartesianProduct(tail, newCurrent);
                output = output.concat(productOfTail);

            } else {
                output.push(newCurrent);

            }
        }
    }
    return output;
}


function objSlice(obj) {
    var slicedObj = angular.copy(obj);  // copy object using angularJs copy method
    delete slicedObj[Object.keys(slicedObj)[0]]; //delete the first key
    return slicedObj;
};

function copy(obj) {
        var res = {};
        for (var p in obj) res[p] = obj[p];
        return res;
    }

console.log(myCartesianProduct(test));

Thanks in advance for your help with this!


请参见以下链接:https://dev59.com/xWct5IYBdhLWcg3wKqQd - Paul
@Paul,这个情况不同。我确实看到了其他帖子(并基于修改创建了代码),但是在这种情况下,我们有嵌套的对象属性,而不是数组的数组。 - Jarnal
是的,我在想也许你可以将Object.keys()应用于子对象,并与另一个问题中的数组笛卡尔积函数相结合,然后将输出从数组的数组重构为对象的数组,例如使用map函数。 - Paul
1个回答

9

好的,让我们从一个生成给定数组乘积的函数开始:

function product(args) {
    if(!args.length)
        return [[]];
    var prod = product(args.slice(1)), r = [];
    args[0].forEach(function(x) {
        prod.forEach(function(p) {
            r.push([x].concat(p));
        });
    });
    return r;
}

下一个例子使用product将类似{a:[1,2], b:[3,4]}的内容转换为[{a:1,b:3},{a:1,b:4},{a:2,b:3},{a:2,b:4}]:
function objectProduct(obj) {
    var keys = Object.keys(obj),
        values = keys.map(function(x) { return obj[x] });

    return product(values).map(function(p) {
        var e = {};
        keys.forEach(function(k, n) { e[k] = p[n] });
        return e;
    });
}

对于您的测试数据,您需要应用两次:

var result = {};
Object.keys(test).forEach(function(k) {
    result[k] = objectProduct(test[k])
});

result = objectProduct(result);

这将给您所需的输出结果。

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接