1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Optimizing R Code with Rcpp
Connected

cvičení

Vážený průměr (verze v C++)

Teď je čas využít nabyté dovednosti a napsat funkci pro výpočet váženého průměru vektoru.

Máš číselný vektor datových hodnot x a číselný vektor vah w. Vážený průměr je součet součinů datových hodnot a vah dělený celkovým součtem vah. Poznámka: vektory x a w musí mít stejný počet prvků.

V R se vážený průměr počítá funkcí weighted.mean(), která provádí výpočet sum(x * w) / sum(w).

Pokyny

100 XP
  • Dokonči definici funkce weighted_mean_cpp().
    • Inicializuj total_xw a total_w na nulu a n na velikost x.
    • Zadej argumenty cyklu for s celým číslem i jako čítačem.
    • Uvnitř cyklu přičti i-tou váhu k total_w a součin i-té datové hodnoty a i-té váhy k total_xw.
    • Vrať celkový součin dělený celkovou vahou.