Array.filter 有多个条件

Question

我有一个对象数组，我想根据条件对其进行过滤和分组。困境来了，因为我有多个条件，并且希望将数组分成几个数组。第一个数组匹配第一个条件，第二个数组匹配第二个条件，...，最后一个数组包含所有不匹配任何条件的对象。

想到这个问题的第一个解决方案是几个 .filter 函数的形式......

var array = [{
  name: 'X',
  age: 18
}, {
  name: 'Y',
  age: 18
}, {
  name: 'Z',
  age: 20
}, {
  name: 'M',
  age: 20
}, {
  name: 'W',
  age: 5
}, {
  name: 'W',
  age: 10
}];
//objects with age 18
var matchedConditional1 = array.filter(function(x){
  return x.age === 18;
});
//objects with age 20
var matchedConditional2 = array.filter(function(x){
  return x.age === 20;
});
//objects with neither age 18 or 20
var matchedNoConditional = array.filter(function(x){
  return (x.age !== 18 && x.age !== 20);
});

但这似乎是多余的，根本无法重用。

于是我修改了Brendan's答案上的函数，得到了这个。

Array.prototype.group = function(f) {
  var matchedFirst = [],
      matchedSecond = [],
      unmatched = [],
      i = 0,
      l = this.length;
  for (; i < l; i++) {
    if (f.call(this, this[i], i)[0]) {
      matchedFirst.push(this[i]);
    } else if (f.call(this, this[i], i)[1]) {
      matchedSecond.push(this[i]);
    } else {
      unmatched.push(this[i]);
    }
  }
  return [matchedFirst, matchedSecond, unmatched];
};
var filteredArray = array.group(function(x){
  return [x.age === 18, x.age === 20];
});

此方法返回一个包含 3 个数组的数组。第一个包含与第一个条件匹配的所有对象，第二个包含与第二个条件匹配的对象，最后一个包含与两者都不匹配的对象。

这种方法的问题在于它仅限于两个条件，因此只有三组对象。这种方法实际上适用于我的特定情况，因为我只有两个条件，但在需要两个以上的情况下是不可重用的。

我希望能够提供尽可能多的条件，并接收该数量的数组以及一个包含不属于任何组的对象的额外数组。

附言。输入和输出不需要是数组，但我认为这更有意义。该方法不必在 .filter 之后建模，它很可能是 .map 函数甚至是 .reduce。任何建议表示赞赏。

编辑：正如@slebetman 所建议的，如果答案允许代码可组合性，那就太好了。

Answer 1

我们将使用findIndex 查找匹配条件的索引，并将该元素放入输出的相应数组元素中：

function makeGrouper(...conditions) {

  return function(array) {
    // Make an array of empty arrays for each condition, plus one.
    var results = conditions.map(_ => []).concat([]);

    array.forEach(elt => {
      var condition = conditions.findIndex(condition => condition(elt));
      if (condition === -1) condition = conditions.length;
      results[condition].push(elt);
    });

    return results;
  };

}

或者，如果你是reduce的粉丝：

function makeGrouper(...conditions) {

  return function(array) {
    return array.reduce((results, elt) => {
      var condition = conditions.findIndex(condition => condition(elt));
      if (condition === -1) condition = conditions.length;
      results[condition].push(elt);
      return results;
    }, conditions.map(_ => []).concat([])));  
  };

}

用法：

const grouper = makeGrouper(
  elt => elt.age === 18,
  elt => elt.age === 20
);

console.log(grouper(data));

此解决方案涉及定义一个函数，您向其提供各种过滤器，该过滤器返回一个函数，然后您可以使用该函数进行实际分组。

Answer 2

试试这样的：

Array.prototype.groups = function(...conditions) {
  return this.reduce(
    (groups, entry) => {
      let indices = [];

      conditions.forEach((cond, i) => {
        if (cond(entry)) indices.push(i);
      });

      if (indices.length === 0) groups[groups.length - 1].push(entry);
      else indices.forEach(i => groups[i].push(entry));

      return groups
    }, 
    Array.apply(null, { length: conditions.length + 1})
      .map(e => [])
  );
}

在这个解决方案中，如果一个条目匹配多个条件，它将出现在相应数量的组中。

使用示例：array.groups(x => x.name === 'X', x => x.age === 18);

最终数组中的最后一个元素 - 不匹配的条目。