获取可变深度数组中的第一个数字数组

时间:2016-09-28 18:42:29

标签: javascript arrays multidimensional-array

我在JavaScript框架中使用一个函数,其返回值可以是以下 ANY

  1. 单个xy坐标对

    [x,y]
    
  2. 一组xy坐标对

    [[x,y],[x,y],...]
    
  3. xy坐标对的数组数组

    [[[x,y],[x,y]],[[x,y],[x,y]],...]
    
  4. 返回值取决于对象的几何形状(单点,线或多行)。无论返回值及其数组深度如何,我都想获取第一个xy坐标对。有效的方法是什么?

    这是我到目前为止实现目标的代码:

    //here is the magic method that can return one of three things :)
    var mysteryCoordinates = geometry.getCoordinates();
    var firstCoord;
    
    if(typeof mysteryCoordinates[0] === 'number') {
        firstCoord = mysteryCoordinates;
    } else if (typeof mysteryCoordinates[0][0] === 'number') {
        firstCoord = mysteryCoordinates[0];
    } else if (typeof mysteryCoordinates[0][0][0] === 'number') {
        firstCoord = mysteryCoordinates[0][0];
    }
    

    我真的很讨厌这个解决方案,我正在寻找更优雅的东西。

2 个答案:

答案 0 :(得分:4)

我想在纯JS中应该这样做;

var    arr = [[[1,2],[1,3]],[[4,8],[3,9]]],
getFirstXY = a => Array.isArray(a[0]) ? getFirstXY(a[0]) : a;

console.log(getFirstXY(arr));

答案 1 :(得分:2)

效率较低但更优雅的解决方案是使用_.flattenhttp://underscorejs.org/#flatten):

let firstCoord = _.flatten(mysteryCoordinates).slice(0, 2);

通过预先切掉前两个元素,你可以平均提高效率:

let firstCoord = _.flatten(mysteryCoordinates.slice(0, 2)).slice(0, 2);

console.log(_.flatten([1,2]).slice(0, 2));
console.log(_.flatten([[1,2],[1,3],[4,8],[3,9]]).slice(0, 2));
console.log(_.flatten([[[1,2],[1,3]],[[4,8],[3,9]]]).slice(0, 2));
<script src="http://underscorejs.org/underscore-min.js"></script>