pythonで辞書型内の各キーに対応する配列の要素をランダムに取り出す方法

Question

pythonか、PyTorchで辞書型内の各キー内の配列の要素を縦軸を決めてランダムに取り出す方法はありますか？
以下のとおり、１発か２発でできるコマンドがあるのか探しています。

dic = {}
dic['a'] = [1,2,3,4] 
dic['b'] = [10,20,30,40]
#・・・

# どのキーの要素も同じ軸で取り出したい
rnd =  関数(dic, 抽出個数=2)
print(rnd['a'])  # [2, 4]
print(rnd['b'])  # [20, 40]

rnd =  関数(dic, 抽出個数=2)
print(rnd['a'])  # [4, 1]
print(rnd['b'])  # [40, 1]

Comments

[takasima20]

ふつうにランダムなインデックスで参照すれば?

[teefpc]

それでは、それぞれのキーでランダムになりませんか？

[takasima20]

ランダムなインデックスのリストを作り、それを参照して a と b から抽出するようにしたらいいんじゃ?

[takasima20]

あー、既存の機能を探しているということですか? これは失礼しました。

Answer 1

• １発か２発でできるコマンドがあるのか探しています。

ありません。

コードを組むなら以下のようになります。

>>> dic = {}
>>> dic['a'] = [1,2,3,4]
>>> dic['b'] = [10,20,30,40]
>>>
>>> import random
>>>
>>> def select(d, n):
...     length = min(len(v) for f in  dic.values())
...     indices = random.sample(range(length), n)
...     return {key:[d[key][i] for i in indices] for key in d}
...
>>> print(select(dic, 2))
{'a': [1, 2], 'b': [10, 20]}
>>> print(select(dic, 2))
{'a': [3, 2], 'b': [30, 20]}
>>> print(select(dic, 2))
{'a': [1, 2], 'b': [10, 20]}
>>> print(select(dic, 2))
{'a': [1, 3], 'b': [10, 30]}

Comments

[teefpc]

これは凄いです。見事なアルゴリズムなので、勉強させていただきます。

Answer 2

1発か2発ということは関数を実装するのではなく既存の機能で実現できるものがないか探しているということでしょうか？
標準だとちょっと思いつきませんがnumpyならキーを配列で指定して抜き出すことで似たような感じにできるんじゃないですかね。
PyTorch使うならnumpyも使ってると思いますが…

dic = {}
dic['a'] = np.array([1,2,3,4])
dic['b'] = np.array([10,20,30,40])
# dicの要素はリストではなくnumpy配列にする

length = len(dic.values()[0]) # 各配列の長さ (4)
cols = np.random.permutation(length)[:num] # ランダムなnum個の縦軸

print(cols) # array([2,4]) (cols例)

print(dic['a'][cols]) # array([2, 4])
print(dic['b'][cols]) # array([20, 40])

print(dic['a'][cols].tolist()) # [2, 4]
print(dic['b'][cols].tolist()) # [20, 40]
    

Comments

[teefpc]

丁度、permutationを使えるか考えていました。ありがとうございました。