使用LINQ获取多维数组的partition (N-1维数组的N-1维实例)

本文关键字:N-1 数组 LINQ 实例 partition 使用 获取 | 更新日期: 2023-09-27 18:05:05

我想从一个3d数组得到一个2d数组。假设我有一个双三维数组A3维度为[10,10,10]
我需要得到一个二维数组A2 = A3[:,5,:],即第二次元的索引等于例如5。

如果我想获得2d数组A2(即2d数组的1d实例)的分区(例如A1=A2[2,:]),我可以这样做(例如第1维的索引设置为2):

double[] A1 = Enumerable.Range(0,A2.Length).Select(x=>A2[2,x]).ToArray();

如何从3维到2维(或者通常从N维到N-1维)?

编辑。输入例子:

double[,,] A3 = new double[2,3,4]{
            {
                {4,3,2,1},
                {5,4,3,2},
                {6,5,4,3}
            },
            {
                {1,2,3,4},
                {2,3,4,5},
                {3,4,5,6}
            }
        };

结果数组A2=A3[:,1,:]必须包含以下内容:{{5,4,3,2},{2,3,4,5}}

使用LINQ获取多维数组的partition (N-1维数组的N-1维实例)

LINQ通常不能很好地处理多维数组。这些操作本质上是单维的。它在嵌套的单维数组中发挥得更好,所以如果你想返回深度为2的锯齿数组,即double[][]而不是double[,], LINQ是更合适的工具。

public static T[][] GetPlane<T>(T[, ,] source, int secondDimensionValue)
{
    return Enumerable.Range(0, source.GetLength(0))
        .Select(i => Enumerable.Range(0, source.GetLength(2))
            .Select(j => source[i, secondDimensionValue, j])
            .ToArray())
        .ToArray();
}

3->2:

static T[][] Collapse<T>(T[][][] array, int y)
{
  return
    Enumerable.Range(0, array.Length).Select(x =>
      Enumerable.Range(0, array[x][y].Length).Select(z =>
        array[x][y][z]
      ).ToArray()
   ).ToArray();
}

如何将这种模式扩展到更高维度的问题应该是相当明显的。

编辑:

以预期的方式扩展到多维数组。如果您对返回多维数组(而不是锯齿数组)感兴趣,那么LINQ可能不是您想要用来解决此问题的工具,因为它没有提供处理多维数组创建的明显或直观的方法。

如果您真的想要多维数组作为输出,我已经包含了一个示例"去毛刺"实现,它应该将其折叠回多维(非锯齿)数组。

static T[][] Collapse<T>(T[,,] array, int y)
{
  return
    Enumerable.Range(0, array.GetLength(0)).Select(x =>
      Enumerable.Range(0, array.GetLength(2)).Select(z =>
        array[x,y,z]
      ).ToArray()
    ).ToArray();
}
static T[,] Deburr<T>(T[][] jagged)
{
  T[,] mArray = new T[jagged.Length, jagged.Max(array => array.Length)];
  foreach (int row in Enumerable.Range(0, jagged.Length))
    foreach (int col in Enumerable.Range(0, jagged[row].Length))
      mArray[row, col] = jagged[row][col];
  return mArray;
}