list-comprehension - 如何在 Chapel 中执行内联 IF 数组构造函数？

Question

我想根据条件创建域的子集。我可以做一个循环，但我想看看是否可以使用内联 if。

只需重新创建阵列 d好像

var d = {1..8};
var e =  [0.875, 0.625, 0.625, 1.0, 0.625, 0.875, 0.625, 0.625];
var p = 0.7;

var vs = for i in d do i;
writeln(" vs: ", vs);

但是，我想提取 d哪里 e[d] < p进入 vs .有没有类似的方法？

vs = [i in d where e[i] < p]
writeln(vs);  // {2,3,5,7,8}

Answer 1

这应该会给你想要的结果:

var vs = for i in d do
           if e[i] < p then i;

请注意 vs是一个数组而不是一个域。如果您想要一个可以使用的域，则应该使用关联域:

var vs : domain(int) = for i in d do
                         if e[i] < p then i;

这个例子会变成这样:

var vs : domain(int);
for i in d {
  if e[i] < p then
    vs.add(i);
}