基于选择掩码的切片数组

本文关键字:掩码 切片 数组 选择 于选择 | 更新日期: 2023-09-27 18:34:22

给定两个数组:

double[] a = new double[]{1.0, 2.0, 3.0};
bool[] b = new bool[]{true, false, true};

有没有一种简单的方法可以根据b选择a?在 R 和其他脚本语言中,您会说:

a[b]

得到{1.0, 3.0}.我无法弄清楚是否有一种干净(不涉及显式循环)的方法在 C# 中执行此操作。也许我应该以不同的方式组织我的数据?

基于选择掩码的切片数组

使用 LINQ Zip 方法,例如:

a.Zip(b, (i, j) => new {i, j}).Where(x => x.j).Select(x => x.i)

您可以使用 LINQ 实现此目的:

double[] a = new double[]{1.0, 2.0, 3.0}; 
bool[] b = new bool[]{true, false, true}; 
var result = a.Where((item, index)=>b[index]);

还有这种更传统的 LINQ 方法:

b.Select((f, i) => f ? i : - 1).Where(i => i != -1).Select(i => a[i]);