对具有索引列表的数组进行切片

Question

假设a = array([1,2,3])和b = array([4,5,6])。我想使用列表对a和b进行切片，并对每个部分执行一些操作，然后将结果返回到数组。例如，我提出了一个虚拟函数来演示用法：

def dummy_function(i):
  A = sum(a[:i])
  B = cumsum(b[i+1:])
  return A*B

例如，此函数将返回dummy_function(2) = 18和dummy_function(1) = 16，但我想使用列表作为其参数来求值：

>>> dummy_function([2,1])
array([18,16])

我不想使用循环来迭代[2,1]的元素，因为我希望它相对较快。我该怎么办？

Answer 1

我不知道我是否正确理解了你想要的东西，但这对我有用：

import numpy as np  


def func(i):
  a = np.array([1,2,3])
  b = np.array([4,5,6])
  A = np.sum(a[:i])
  B = np.cumsum(b[i:])
  C = A*B
  return C[0]

print(func(2))

结果为18