更绝妙的列表操作方式

Question

我有两个这样的列表：

def a = [100,200,300]
def b = [30,60,90]

我想要像这样操作 a 的 Groovier 方式：

1) a 的第一个元素应该更改为 a[0] -2*b[0]

2）a的第二个元素应更改为a[1]-4*b[1]

3）第三个元素a 应该更改为 a[2]-8*b[2]

（前提是 a 和 b 的长度与 >3)

如果列表像这样更改为 map，我们可以说：

def a1 = [100:30, 200:60, 300:90]

在这种情况下如何执行与上述相同的操作。

提前致谢。

Answer 1

对于List，我会选择：

def result = []
a.eachWithIndex{ item, index ->
  result << item - ((2**index) * b[index])
}

对于Map，它更容易一些，但仍然需要外部状态：

int i = 1
def result = a.collect { k, v -> k - ((2**i++) * v) }

遗憾的是，Groovy 没有 zip，在本例中 - 类似于 zipWithIndex 或collectWithIndex。

Answer 2

使用收集

响应 Victor 的评论，您可以使用收集来完成此操作

def a = [100,200,300]
def b = [30,60,90]

// Introduce a list `c` of the multiplier
def c = (1..a.size()).collect { 2**it }

// Transpose these lists together, and calculate
[a,b,c].transpose().collect { x, y, z ->
  x - y * z
}

使用注入

您还可以使用注入，传入乘数和结果的映射，然后在最后取出结果：

def result = [a,b].transpose().inject( [ mult:2, result:[] ] ) { acc, vals ->
  acc.result << vals.with { av, bv -> av - ( acc.mult * bv ) }
  acc.mult *= 2
  acc
}.result

类似地，您可以对映射使用注入：

def result = a1.inject( [ mult:2, result:[] ] ) { acc, key, val ->
  acc.result << key - ( acc.mult * val )
  acc.mult *= 2
  acc
}.result

使用 inject 的优点是不需要声明外部变量，但缺点是难以阅读代码（正如 Victor 指出的那样）在注释中，这对代码进行了静态分析对于 IDE 和 groovypp 来说很难甚至不可能）

Answer 3

def a1 = [100:30, 200:60, 300:90]

a1.eachWithIndex{item,index ->
println item.key-((2**(index+1))*item.value)
i++
}