Issue #309, adding symmetric difference as _.symDifference, and making _.difference accept any number of arguments.

This commit is contained in:
Jeremy Ashkenas
2011-11-23 12:42:22 -05:00
parent 65ab040a7b
commit 7653c5fe09
2 changed files with 20 additions and 3 deletions

View File

@@ -110,6 +110,14 @@ $(document).ready(function() {
test("arrays: difference", function() { test("arrays: difference", function() {
var result = _.difference([1, 2, 3], [2, 30, 40]); var result = _.difference([1, 2, 3], [2, 30, 40]);
equals(result.join(' '), '1 3', 'takes the difference of two arrays'); equals(result.join(' '), '1 3', 'takes the difference of two arrays');
var result = _.difference([1, 2, 3, 4], [2, 30, 40], [1, 11, 111]);
equals(result.join(' '), '3 4', 'takes the difference of three arrays');
});
test("arrays: symDifference", function() {
var result = _.symDifference([1, 2, 3], [2, 22, 222], [3, 33, 333], [222, 333, 444], [5]);
equals(result.join(' '), '1 22 33 444 5', 'takes the symmetric difference');
}); });
test('arrays: zip', function() { test('arrays: zip', function() {

View File

@@ -402,10 +402,19 @@
}); });
}; };
// Take the difference between one array and another. // Take the difference between one array and a number of other arrays.
// Only the elements present in just the first array will remain. // Only the elements present in just the first array will remain.
_.difference = function(array, other) { _.difference = function(array) {
return _.filter(array, function(value){ return !_.include(other, value); }); var rest = _.flatten(slice.call(arguments, 1));
return _.filter(array, function(value){ return !_.include(rest, value); });
};
// Take the symmetric difference between a list of arrays. Only the elements
// present in one of the input arrays will remain.
_.symDifference = function() {
return _.reduce(arguments, function(memo, array) {
return _.union(_.difference(memo, array), _.difference(array, memo));
});
}; };
// Zip together multiple lists into a single array -- elements that share // Zip together multiple lists into a single array -- elements that share