通过过滤函数划分数组

时间:2012-07-30 23:22:05

标签: javascript

我有一个Javascript数组,根据每个元素调用的函数是返回true还是false,我想将其拆分为两个。从本质上讲,这是一个array.filter,但我还想了解过滤 out 的元素。

目前,我的计划是使用array.forEach并在每个元素上调用谓词函数。根据这是真还是假,我将当前元素推送到两个新数组中的一个。是否有更优雅或更好的方法来做到这一点?一个array.filter,例如,在它返回false之前,它会将元素推送到另一个数组上吗?

12 个答案:

答案 0 :(得分:29)

使用ES6,您可以使用reduce:

来使用扩展语法
function partition(array, isValid) {
  return array.reduce(([pass, fail], elem) => {
    return isValid(elem) ? [[...pass, elem], fail] : [pass, [...fail, elem]];
  }, [[], []]);
}

const [pass, fail] = partition(myArray, (e) => e > 5);

或单行:

const [pass, fail] = a.reduce(([p, f], e) => (e > 5 ? [[...p, e], f] : [p, [...f, e]]), [[], []]);

答案 1 :(得分:23)

您可以使用lodash.partition

var users = [
  { 'user': 'barney',  'age': 36, 'active': false },
  { 'user': 'fred',    'age': 40, 'active': true },
  { 'user': 'pebbles', 'age': 1,  'active': false }
];

_.partition(users, function(o) { return o.active; });
// → objects for [['fred'], ['barney', 'pebbles']]

// The `_.matches` iteratee shorthand.
_.partition(users, { 'age': 1, 'active': false });
// → objects for [['pebbles'], ['barney', 'fred']]

// The `_.matchesProperty` iteratee shorthand.
_.partition(users, ['active', false]);
// → objects for [['barney', 'pebbles'], ['fred']]

// The `_.property` iteratee shorthand.
_.partition(users, 'active');
// → objects for [['fred'], ['barney', 'pebbles']]

ramda.partition

R.partition(R.contains('s'), ['sss', 'ttt', 'foo', 'bars']);
// => [ [ 'sss', 'bars' ],  [ 'ttt', 'foo' ] ]

R.partition(R.contains('s'), { a: 'sss', b: 'ttt', foo: 'bars' });
// => [ { a: 'sss', foo: 'bars' }, { b: 'ttt' }  ]

答案 2 :(得分:14)

这听起来与Ruby's Enumerable#partition方法非常相似。

如果函数不能有副作用(也就是说,它不能改变原始数组),那么分割数组的效率要高于迭代每个元素并将元素推送到你的两个元素之一阵列。

话虽如此,在Array上创建一个方法以执行此功能可能更“优雅”。在此示例中,过滤器函数在原始数组的上下文中执行(即,this将是原始数组),并且它接收元素和元素的索引作为参数(类似于{{3 }}):

Array.prototype.partition = function (f){
  var matched = [],
      unmatched = [],
      i = 0,
      j = this.length;

  for (; i < j; i++){
    (f.call(this, this[i], i) ? matched : unmatched).push(this[i]);
  }

  return [matched, unmatched];
};

console.log([1, 2, 3, 4, 5].partition(function (n, i){
  return n % 2 == 0;
}));

//=> [ [ 2, 4 ], [ 1, 3, 5 ] ]

答案 3 :(得分:6)

在过滤功能中,您可以将假项目推送到函数外的另一个变量:

var bad = [], good = [1,2,3,4,5];
good = good.filter(function (value) { if (value === false) { bad.push(value) } else { return true});

当然value === false需要真正的比较;)

但它几乎完成了与forEach相同的操作。我认为你应该使用forEach来提高代码的可读性。

答案 4 :(得分:6)

您可以使用reduce:

function partition(array, callback){
  return array.reduce(function(result, element, i) {
    callback(element, i, array) 
      ? result[0].push(element) 
      : result[1].push(element);

        return result;
      }, [[],[]]
    );
 };

更新。使用ES6语法,您也可以使用递归:

function partition([current, ...tail], f, [left, right] = [[], []]) {
    if(current === undefined) {
        return [left, right];
    }
    if(f(current)) {
        return partition(tail, f, [[...left, current], right]);
    }
    return partition(tail, f, [left, [...right, current]]);
}

答案 5 :(得分:3)

试试这个:

function filter(a, fun) {
    var ret = { good: [], bad: [] };
    for (var i = 0; i < a.length; i++)
        if (fun(a[i])
            ret.good.push(a[i]);
        else
            ret.bad.push(a[i]);
    return ret;
}

DEMO

答案 6 :(得分:2)

我想出了这个小家伙。它用于你描述的每一个和所有,但在我看来它看起来干净简洁。

&#13;
&#13;
//Partition function
function partition(array, filter) {
  let pass = [], fail = [];
  array.forEach((e, idx, arr) => (filter(e, idx, arr) ? pass : fail).push(e));
  return [pass, fail];
}

//Run it with some dummy data and filter
const [lessThan5, greaterThanEqual5] = partition([0,1,4,3,5,7,9,2,4,6,8,9,0,1,2,4,6], e => e < 5);

//Output
console.log(lessThan5);
console.log(greaterThanEqual5);
&#13;
&#13;
&#13;

答案 7 :(得分:2)

易于阅读。

const partition = (arr, condition) => {
    const trues = arr.filter(el => condition(el));
    const falses = arr.filter(el => !condition(el));
    return [trues, falses];
};

// sample usage
const nums = [1,2,3,4,5,6,7]
const [evens, odds] = partition(nums, (el) => el%2 == 0)

答案 8 :(得分:2)

这里很多答案都使用String equals()来构建可变的累加器,并正确地指出,对于大型数组,这比使用散布运算符在每次迭代中复制新数组的效率更高。缺点是,使用短的lambda语法,它不如“纯”表达式好。

但是一种解决方法是使用逗号运算符。在类似C的语言中,逗号是始终返回右手操作数的运算符。您可以使用它来创建一个调用void函数并返回值的表达式。

Array.prototype.reduce

如果您利用布尔表达式隐式转换为0和1的数字这一事实,虽然我认为它不那么可读,但您可以使其更加简洁:

function partition(array, predicate) {
    return array.reduce((acc, item) => predicate(item)
        ? (acc[0].push(item), acc)
        : (acc[1].push(item), acc), [[], []]);
}

用法:

function partition(array, predicate) {
    return array.reduce((acc, item) => (acc[+!predicate(item)].push(item), acc), [[], []]);
}

答案 9 :(得分:1)

那呢?

[1,4,3,5,3,2].reduce( (s, x) => { s[ x > 3 ].push(x); return s;} , {true: [], false:[]} )

这可能比散布运算符更有效

或更短,但更丑

[1,4,3,5,3,2].reduce( (s, x) => s[ x > 3 ].push(x)?s:s , {true: [], false:[]} )

答案 10 :(得分:0)

我最终这样做是因为它易于理解(并使用打字稿完全键入)。

const partition = <T>(array: T[], isValid: (element: T) => boolean): [T[], T[]] => {
  const pass: T[] = []
  const fail: T[] = []
  array.forEach(element => {
    if (isValid(element)) {
      pass.push(element)
    } else {
      fail.push(element)
    }
  })
  return [pass, fail]
}

// usage
const [pass, fail] = partition([1, 2, 3, 4, 5], (element: number) => number > 3)

答案 11 :(得分:0)

单行分区

const partition = (a,f)=>a.reduce((p,q)=>(p[+!f(q)].push(q),p),[[],[]]);

演示

// to make it consistent to filter pass index and array as arguments
const partition = (a, f) =>
    a.reduce((p, q, i, ar) => (p[+!f(q, i, ar)].push(q), p), [[], []]);

console.log(partition([1, 2, 3, 4, 5], x => x % 2 === 0));
console.log(partition([..."ABCD"], (x, i) => i % 2 === 0));

对于打字稿

const partition = <T>(
  a: T[],
  f: (v: T, i?: number, ar?: T[]) => boolean
): [T[], T[]] =>
  a.reduce((p, q, i, ar) => (p[+!f(q, i, ar)].push(q), p), [[], []]);