在多维数组中查找值并返回所有项目c#

时间:2011-04-10 14:37:11

标签: c# .net arrays

这是问题,我有一个如下定义的数组:

int[,] Users = new int[1000,3];

它的数据类似于:

  • 0,1,2
  • 1,2,1
  • 2,3,2
  • 3,3,4
  • 4,2,3 ...

我的脚本根据需要使用该数组。但我需要能够根据其中一个维度过滤数组,并返回所有可用的匹配项。

例如,过滤维度[1],希望所有匹配的“3” 将返回一个包含以下内容的数组:

  • 2,3,2
  • 3,3,4-

有人能帮我一把吗?

非常感谢。

1 个答案:

答案 0 :(得分:10)

如果您可以将数组从int[,]更改为int[][],那么您可以使用LINQ轻松实现此目的。

  int[][] users = new int[][]
  {
    new int[]{0,1,2},
    new int[]{1,2,1},
    new int[]{2,3,2},
    new int[]{3,3,4},
    new int[]{4,2,3}
  };


  var result = from u in users 
               where u[1] == 3 
               select u;

如果无法更改数组,则可以编写Filter函数,如下所示。

public static IEnumerable<T[]> Filter<T>(T[,] source, Func<T[], bool> predicate)
{
  for (int i = 0; i < source.GetLength(0); ++i)
  {
    T[] values = new T[source.GetLength(1)];
    for (int j = 0; j < values.Length; ++j)
    {
      values[j] = source[i, j];
    }
    if (predicate(values))
    {
      yield return values;
    }
  }      
}

然后可以按照以下方式调用上述内容

var result = Filter(users, u => u[1] == 3);

您可以更进一步,为Where函数实现自己的自定义Linq扩展,以便过滤T[,]数组。这是一个可以帮助你入门的天真例子。

public static class LinqExtensions
{
  public static IEnumerable<T[]> Where<T>(this T[,] source, Func<T[], bool> predicate)
  {
    if (source == null) throw new ArgumentNullException("source");
    if (predicate == null) throw new ArgumentNullException("predicate");
    return WhereImpl(source, predicate);
  }

  private static IEnumerable<T[]> WhereImpl<T>(this T[,] source, Func<T[], bool> predicate)
  {      
    for (int i = 0; i < source.GetLength(0); ++i)
    {
      T[] values = new T[source.GetLength(1)];
      for (int j = 0; j < values.Length; ++j)
      {
        values[j] = source[i, j];
      }
      if (predicate(values))
      {
        yield return values;
      } 
    }
  }    
}

使用此功能,您可以再次使用Linq,如第一个示例中那样过滤数组

  var result = from u in users 
               where u[1] == 3 
               select u;