回答編集履歴
2
修正
answer
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
1発か2発ということは関数を実装するのではなく既存の機能で実現できるものがないか探しているということでしょうか？
|
|
2
|
-
標準だとちょっと思いつきませんがnumpyならキーを配列
|
|
2
|
+
標準だとちょっと思いつきませんがnumpyならキーを配列で指定して抜き出すことで似たような感じにできるんじゃないですかね。
|
|
3
3
|
PyTorch使うならnumpyも使ってると思いますが…
|
|
4
4
|
|
|
5
5
|
```py
|
1
修正
answer
CHANGED
|
@@ -10,8 +10,9 @@
|
|
|
10
10
|
|
|
11
11
|
length = len(dic.values()[0]) # 各配列の長さ (4)
|
|
12
12
|
cols = np.random.permutation(length)[:num] # ランダムなnum個の縦軸
|
|
13
|
-
# 例えば cols = np.array([2, 4])
|
|
14
13
|
|
|
14
|
+
print(cols) # array([2,4]) (cols例)
|
|
15
|
+
|
|
15
16
|
print(dic['a'][cols]) # array([2, 4])
|
|
16
17
|
print(dic['b'][cols]) # array([20, 40])
|
|
17
18
|
|