如何理解 JavaScript 中 Array 的 map/filter 函数

最近一直想写一系列文章,好好介绍一下Javascript下的函数编程,以及 ReactiveX 这个神器。拖延症犯了,连提纲都没规划出来。不过真心想聊这个话题,今天就先聊一聊一些简单的小技巧。

如何理解 Array 的 map 函数 ?

我们先来看一个简单的例子,假设有一个数组 [1,2,3,4],如果想对里面所有的数全部加上10,那么最普通的做法就是:

var input = [1,2,3,4];
var output = [];
for(var i = 0; i < input.length; i++ ) {
    var newData = input[i] + 10 ;
    output.push(newData);
}
console.log(output);

再来看另外一个稍微复杂一点的例子,假如我们从网路读取到一批数据,然后把他们渲染到页面上:

var dataList = getJsonFromRemote();
var output = [];
for(var i = 0; i < dataList.length; i++ ) {
    var el = '
  • ' + dataList[i] + '
  • ' output.push(el); } console.log(output);

    我们来观察一下这两个例子有什么样的共性?他们都是从一个数组中循环读出每一个数据,然后通过计算后得到一个新数据,然后放入到一个数组中,那我们来对代码进行一些优化:

    function add(data) {
        return data + 10;
    }
    
    var input = [1,2,3,4];
    var output = [];
    for(var i = 0; i < input.length; i++ ) {
        var newData = add(input[i]);
        output.push(newData);
    }
    console.log(output);
    
    function renderLi(data) {
        return '
  • ' + dataList[i] + '
  • '; } var input = getJsonFromRemote(); var output = []; for(var i = 0; i < input.length; i++ ) { var newData = renderLi(input[i]); output.push(el); } console.log(output);

    我们把这两个代码片段中的循环逻辑修改成了一个小函数,大家再对比一下这两个for循环内部,几乎只有那个“小函数”的名字不一样,其他部分一模一样,我们给这个“小函数”一个好听的名字——算法。

    对,我们在对一个数组进行循环处理时,大部分情况下,如果他们的输入是一个合法的数组,输出也是一个合法的数组,唯一不同的只是他们的“算法”不一样而已。于是,我们再来优化一下代码:

    function map(array, func) {
        var output = [];
        for(var i = 0; i < array.length; i++ ) {
            output.push(func(array[i]));
        }
        return output;
    }
    
    function add(data) {
        return data + 10;
    }
    
    function renderLi(data) {
        return '
  • ' + data[i] + '
  • '; } var input1 = [1,2,3,4]; var output1 = map(input1, add); var input2 = getJsonFromRemote(); var output2 = map(input2, renderLi);

    我们把一个 for 循环实现成了一个 map 函数,它接受两个参数:一个是数组,一个是算法函数。于是在我们下面的调用中,就再也看不到 for 循环的代码了。

    目前的 map 函数接受两个参数,跟我们最开始提到的 Array 对象的 map 函数还稍微有一点点差异,我们再调整一次:

    class Array {
        constructor() {
            this._list = [];
        }
    
        map(func) {
            var output = [];
            for(var i = 0; i < this._list.length; i++ ) {
                output.push(func(array[i]));
            }
            return output;
        }
    }
    

    我们把一个普通的 map 函数放到 Array 类上,那么它也只需要接受一个参数了,就是“算法”。所以我们的逻辑代码也相应调整为:

    [1,2,3,4].map(function(item) {
        return item + 10;
    });
    
    getJsonFromRemote().map(function(item) {
        return '
  • ' + item[i] + '
  • '; });

    如何理解 Array 的 filter 函数 ?

    如果 for 循环以后都变成了 map 函数调用的话,那么大家一定会有一个疑问:如果并不是想对一个数组进行“算法变化”,而只是想获取原数组中“符合规格”的那部分数据,该怎么做呢?

    这是就该 filter 出场了,看名字就可以知道,它就是一个过滤器。假设有一个 number 数组,我们只想获取它的偶数。我们先还是来看看传统的“笨办法”:

    var input = [1, 2, 3, 4, 5, 6, 7, 8, 9];
    var output = [];
    for(var i = 0; i < input.length; i++ ) {
        if(input[i] % 2 == 0) {
            output.push(input[i]);
        }
    }
    console.log(output);
    

    然后我们在用学习 map 时的办法,来对他进行一下优化:

    function filter(input, func) {
        var output = [];
        for(var i = 0; i < input.length; i++ ) {
            if(func(input[i])) {
                output.push(input[i]);
            }
        }
        return output;
    }
    
    function even(data) {
        return input[i] % 2 == 0;
    }
    
    var input = [1, 2, 3, 4, 5, 6, 7, 8, 9];
    var output = filter(input, even);
    console.log(output);
    

    可以看到 filter 函数和 map 函数的实现很相似,主体依然是 for 循环,不过循环体的内部是先调用算法计算数据,然后依据计算结果来选择是否把当前数据复制到 output 中,结果为 true 则保留,为 false 则过滤掉。even 函数的作用就是对传入的数据进行判断,如果是偶数,则返回 true。最终的版本如下:

    [1, 2, 3, 4, 5, 6, 7, 8, 9].filter(function(item) {
        return item[i] % 2 == 0;
    });
    

    综合范例

    我们接下来来实现一个非常真实的业务逻辑:我们用 nodejs 读取一个文件夹下所有的第一层子文件夹的名字,然后转化为绝对路径(以下代码暂时不考虑错误逻辑)。

    var fs = require('fs');
    var path = require('path');
    
    var dirPath = '...';
    var contentList = fs.readdirSync(dirPath);  //读取文件夹下的所有内容
    var output = [];
    for(var i = 0; i < contentList.length; i++) {
        var name = contentList[i];
        var subDirPath = path.resolve(dirPath, name); // 获取绝对路径
        if(fs.statSync(subDirPath).isDirectory()) {  // 判断是否为文件夹,排除掉文件
            output.push(fullPath);
        }
    }
    

    再来看看优化实现:

    var fs = require('fs');
    var path = require('path');
    
    var dirPath = '...';
    var output = fs.readdirSync(dirPath).map(function(item){
        return path.resolve(dirPath, item);
    }).filter(function(item) {
        return fs.statSync(item).isDirectory();
    });
    

    再换成 lambda 版本看看?

    var fs = require('fs');
    var path = require('path');
    
    function getSubDirList(dirPath) {
        return fs.readdirSync(dirPath)
            .map(item => path.resolve(dirPath, item))
            .filter(item => fs.statSync(item).isDirectory());
    }
    
    var dirPath = '...';
    var output = getSubDirList(dirPath);
    

    怎么样? 有没有很清爽的感觉?不过我更喜欢下面这种美妙的写法:

    var R = require('ramda');
    var fs = require('fs');
    var path = require('path');
    
    var getSubDirList = R.compose(
        R.filter(item => fs.statSync(item).isDirectory()),
        R.map(item => path.resolve(dirPath, item)),
        fs.readdirSync
    );
    
    var dirPath = '...';
    var output = getSubDirList(dirPath);
    

    最后这段代码有人能看明白么?卖个关子,下次再讲。

    你可能感兴趣的:(如何理解 JavaScript 中 Array 的 map/filter 函数)