使用下划线库替代_.invert()

Question

我需要更改现有的映射，将键交换为值，将值转换为键。由于映射中有重复的值，所以不能使用下划线库的_.invert()。

function map() {
        return {
             'eatables': {
                 apple: 'fruits',
                 orange: 'fruits',
                 guava: 'fruits',
                 brinjal: 'vegetables',
                 beans: 'vegetables',
                 rose: 'flowers',
             }
        }
    }

    var reverseMap = _.invert(map()['eatables']); 
// invert function works for distinct values.
    console.log (reverseMap); 
// which is giving Object {fruits: "guava", vegetables: "brinjal",flowers:"rose"}

但我希望输出结果

Object {fruits: ["apple","orange","guava"], vegetables: ["brinjal","beans"], flowers:"rose"}

我试过如下所示，我只是坚持如何找到映射值是不同的还是多个？

   var newObj = invert(map()['eatables']);
_.each(newObj, function(key) {
    if (Array.isArray(key)) {
        _.each( key, function(value) {
            console.log(value);
        });
    } else {
        console.log("else:"+key);
    }
});

function invert(srcObj) {
    var newObj = {};
    _.groupBy(srcObj, function(value, key ) {
        if (!newObj[value]) newObj[value] = []; //Here every thing is array, can i make it string for values which are unique.
        newObj[value].push(key);
    });
    return newObj;
}

让我选择使用下划线库。

Answer 1

您可以使用此函数。该函数使用Object.keys生成一个数组，该数组包含输入中传递的对象的键。然后，它访问原始对象的值，并将它们用作新对象中的键。当两个值映射到同一个键时，它将它们推入数组中。

   function invert(obj) {
    var result = {};
    var keys = Object.keys(obj);
    for (var i = 0, length = keys.length; i < length; i++) {
        if (result[obj[keys[i]]] instanceof Array) {
            result[obj[keys[i]]].push(keys[i])
        } else if (result[obj[keys[i]]]) {
            var temp = result[obj[keys[i]]];
            result[obj[keys[i]]] = [temp, keys[i]];
        } else {
            result[obj[keys[i]]]=keys[i];
        }
    }
    return result;
}

https://jsfiddle.net/6f2ptxgg/1/

Answer 2

可以使用下划线每一个迭代数据并将结果推送到数组中。它会给你预期的产量。

function customInvert(data) {
    var result = {};
    _.each(data, function (value, key) {
        if (_.isUndefined(result[value])) {
            result[value] = key;
        } else if(_.isString(result[value])) {
           result[value] = [result[value], key];
        } else {
           result[value].push(key)
        }            
    });
    return result;
}

customInvert({
   apple: 'fruits',
   orange: 'fruits',
   guava: 'fruits',
   brinjal: 'vegetables',
   beans: 'vegetables',
   rose: 'flowers',
 })