这是问题,我有一个如下定义的数组:
int[,] Users = new int[1000,3];
它的数据类似于:
我的脚本根据需要使用该数组。但我需要能够根据其中一个维度过滤数组,并返回所有可用的匹配项。
例如,过滤维度[1],希望所有匹配的“3” 将返回一个包含以下内容的数组:
有人能帮我一把吗?
非常感谢。
答案 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;