pythonでの加重平均について

Question

###前提・実現したいこと
下記のような配列があり、amountの上位n人目までの加重平均を取りたいです。
10人目までの加重平均を取りたい場合は例の配列で言うと(5003+4007)/10ということをしたいです。
また、scoreとamountはHTTPリクエストから返ってくる数字を整形したものなのでリクエスト毎に値が違います。
下記の配列は一例ですが、どのような値が返ってきてもn番目までを加重平均の取れるようにしたいです。

###返ってくる値の一例

score = [500, 400, 300, 200, 100, 50, 0]#ここは点数で降順に並ぶ。
amount = [3, 8, 3, 5, 10, 3, 1]#人数

下記のようにすれば、全体の加重平均は取れるのですが、n人目までの加重平均を取り方が分からず困っています。
n人目というのはamountの上位n人目という意味で、amountの左から上位n人までのことです。

wt_avg = sum([s*a for s,a in zip(score,amount)]) / sum(amount)#234.84848484848484

scoreは降順で並んでいて、amountはランダムの数字になります。
初歩的な質問で恐縮ですが、よろしくお願いします。

###補足情報(言語/FW/ツール等のバージョンなど)
python3.5.2

Answer 1

最初にデータを平たく並べるところがミソです。

def wighted_avg(score, amount, top_n):
    liner = sum(([s] * a for s, a in zip(score, amount)), [])
    return sum(liner[:top_n]) / top_n

score = [500, 400, 300, 200, 100, 50, 0]
amount = [3, 8, 3, 5, 10, 3, 1]
print(wighted_avg(score, amount, 10))

以下は別解。itertools祭り版。

from itertools import chain
from itertools import islice
from itertools import repeat

def wighted_avg(score, amount, top_n):
    flatten = chain.from_iterable(map(repeat, score, amount))
    return sum(islice(flatten, top_n)) / top_n

Comments

[jaja]

解決しました！ありがとうございます。
このような関数を作ればいけるのですね。
誠にありがとうございます！

[mkgrei]

失礼しました。
質問の意図を勘違いしていました。

Answer 2

デバッグしているうちに解決してしまいましたが…

ここによると、sumによるflattenは遅いそうです。
当該記述を参考に私が書いてみたのが下のコードです。

score = [500, 400, 300, 200, 100, 50, 0]
amount = [3, 8, 3, 5, 10, 3, 1]

num_of_human = 10

from itertools import chain
# http://d.hatena.ne.jp/xef/20121027/p2
target = list(chain.from_iterable([[s]*a for s,a in zip(score,amount)]))[0:num_of_human]

avg = sum(target)/len(target)
print(avg)

Comments

[YouheiSakurai]

chainだと上手く描けないなぁと思ってreduce的にsumを使いました。まぁsumだと遅いと思います。chainのfrom_iterable、勉強になります。

Answer 3

すみません、修正しました。慌てて作ったのでちょっと良くないコードですね...

score = [500, 400, 300, 200, 100, 50, 0]
amount = [3, 8, 3, 5, 10, 3, 1]

num_of_human = 10

score_total = 0
human_total = 0
for s, a in zip(score, amount):
    for h in range(a):
        score_total += s
        human_total += 1
        if human_total >= num_of_human:
            break
    if human_total >= num_of_human:
        break

wt_avg = score_total / num_of_human            

Comments

[jaja]

回答ありがとうございます。
amountのn番目というかamoutの上位n人の加重平均が取りたいのですが、
n番目と質問したのが悪かったです。。。
質問が分かりにくくて申し訳ないです。

[LouiS0616]

しっかり読めば理解できた質問内容ではありました。大変失礼しました。
慌てて作ったのでかなり汚いコードですが、いちおう要件を満たすように編集しました。

[LouiS0616]

もうちょっときれいに書けるはずなので、出来次第追記します。

[jaja]

追加の回答ありがとうございます！
n人目までの加重平均が取れました。ありがとうございます。
宜しければ、追記も頂ければ嬉しいです。

[LouiS0616]

YouheiSakuraiさんのコードが非常に美しいですので、そちらを見ていただいた方がよさそうです。

[jaja]

回答とコメントありがとうございました。
高評価させていただきました。

[LouiS0616]

ご丁寧にありがとうございます、慌ててよろしくない回答をしてしまってすみません。
焼け石に水レベルですが、上記ループはfor-else文を用いるとちょっときれいになります。
さらに改良出来そうですが、もともとの方針的にYouheiSakuraiさんのコードの簡潔さは超えられなさそうですので、ちょっと控えておきます。