listから重複なしで重みをつけてランダムで複数選びたい

Question

実行したいこと

python3.9で以下のようなlist(date)から一つ一つの要素を選ぶ重みのlist(plist)を使って重みをつけてn個取り出そうとしています。
random.sample()だと重みをつけられず、random.choice()だと重複を許してしまいます。
どのようにすればよいのでしょうか。

使うデータ(例)

[a,b,c,d,e,f,g,h,i,j,k,l,m,n]

[1,0.5,0.5,1,1,1,0.5,0.5,0.5,1,1,1,1,1]

Comments

[8524ba23]

n < dateの要素数 という条件のもと、ということでしょうか？
n のほうが大きいと重複せざるを得ないので。

[ppaul]

date = ['a','b','c']
plist = [1, 10, 10]
のとき、どのような結果が出ることを期待していますか？

[orange-gummy]

n<dateの要素数　という条件のもとです

[orange-gummy]

date = ['a','b','c']
plist = [1, 10, 10]
のときn=1の場合は1/21でa、10/21でb、10/21でcを得たいです
n=2の場合はn=1と同じように一つの要素を得て、
それがaの場合は10/20でbかc、
それがbの場合は1/11でa、10/11でc、
それがcの場合は1/11でa、10/11でｂを得たいです

Answer 1

can110さんの回答の改良版です。
ランダム複数選択を何度もやりたい場合ならば、元のリストをなるべく壊さないほうが良いでしょう。

import random

date = list('abcdefghijklmn')
plist = [1,0.5,0.5,1,1,1,0.5,0.5,0.5,1,1,1,1,1]
indices = range(len(date))
plist_copy = plist.copy()

n = 5
ret = []
for i in range(n):
    idx = random.choices(indices, plist_copy)[0]
    ret.append(date[idx])
    plist_copy[idx] = 0

実行結果

>>> print(ret)
['h', 'j', 'a', 'm', 'c']

Comments

[8524ba23]

重みが０を含む場合の挙動が私の回答と異なりますね。
choicesでは重みが０の要素も選択されえるので。
極端なケースだとすべての重みが０の場合。

このchoicesの挙動が質問者の意図に合っているか次第ですが
合っていなければ、重み０は最初から取り除くなど私の回答も修正する必要ありますね。

[orange-gummy]

回答ありがとうございます！いろいろなやり方があるのですね
しかし、今回自分のやりたいことはcan110さんの方法ですることができたのでcan110さんをベストアンサーにさせていただきます！

Answer 2

• 位置ベースで処理
• random.choicesで重み付きで１つの要素を選択
• 選択された要素＋重みを取り除き、次の処理へ

といった感じでよいかと思います。

import random

date = list('abcdefghijklmn')
plist = [1,0.5,0.5,1,1,1,0.5,0.5,0.5,1,1,1,1,1]

n = 5
ret = []
for i in range(n):
    idx = random.choices(range(len(date)), plist)[0]
    ret.append(date.pop(idx))
    plist.pop(idx)

print(ret) # ['m', 'b', 'a', 'f', 'd']

Comments

[orange-gummy]

自分のしたいようにできました！ありがとうございます！

Answer 3

いろいろやり方は考えられますが、例えば、choiceで重みを指定して1つずつ選び、選んだものをその都度元のリストから削除して、必要な回数だけ繰り返せば良いかと。