python - Python - これをベクトル形式で記述する方法

Question

def compute(c, r):
    s = 0;
    l = len(c);
    for i in range(l):
        s = s + c[i]*f(r[i]);
    return s

Answer 1

ベクトル形式で何を意味するのかわかりませんが(numpyを使用していない限り?)、関数を次のように記述します。

def compute(c, r):
  return sum(x*f(y) for x,y in zip(c,r))

numpy を使用している場合は、ジェネレーター式の代わりに配列全体の式を使用できますが、その場合は numpy 配列cで なければなりません。r

def compute(c, r):
  return (c*f(r)).sum()