pythonでの配列の初期化

Question

前提・実現したいこと

配列の初期化で以下のコードで3*2の配列であるRanの定義の部分が長いのでもっと短くスマートに書く方法はないでしょうか？

該当のソースコード

import numpy as np
import random

Min =np.array([[3, 4],[7, 8],[11, 12]])
Max =np.array([[6, 7],[10, 11],[14, 15]])

Ran =np.array([[random.randint(Min[0, 0], Max[0, 0]), random.randint(Min[0, 1], Max[0, 1])],[random.randint(Min[1, 0], Max[1, 0]), random.randint(Min[1, 1], Max[1, 1])],[random.randint(Min[2, 0], Max[2, 0], random.randint(Min[2, 1], Max[2, 1]]])

Answer 1

解決済みですが、numpy.vecotorizeでもいけそうです。

import numpy as np
import random

Min = np.array([[3, 4], [7, 8], [11, 12]])
Max = np.array([[6, 7], [10, 11], [14, 15]])

@np.vectorize
def initialize(min_, max_):
    return random.randint(min_, max_)

Ran = initialize(Min, Max)

Answer 2

これってどの成分も取りうる値が4種類なので

np.random.randint(0, 4, Min.shape) + Min

でよくないですか？

Comments

[nekohouse]

ありがとうございます
質問のためなるべく簡略化して書いたのですが実際のコードだともう少し複雑でとりうる値の幅もバラバラです
書き方が悪かったです、すいません

Answer 3

たとえば内包表記つかってみて以下のような変更でどうでしょうか。

import numpy as np
import random

Min =np.array([[3, 4],[7, 8],[11, 12]])
Max =np.array([[6, 7],[10, 11],[14, 15]])

# 元のコード
Ran = np.array([
    [random.randint(Min[0, 0], Max[0, 0]), random.randint(Min[0, 1], Max[0, 1])],
    [random.randint(Min[1, 0], Max[1, 0]), random.randint(Min[1, 1], Max[1, 1])],
    [random.randint(Min[2, 0], Max[2, 0]), random.randint(Min[2, 1], Max[2, 1])]
    ])
print(Ran)
# [[ 3  6]
#  [ 9  9]
#  [13 13]]

# 内包表記
Ran_2 = np.array([[random.randint(Min[i, 0], Max[i, 0]), random.randint(Min[i, 1], Max[i, 1])] for i in range(3)])
print(Ran_2)
# [[ 3  4]
#  [ 7  9]
#  [12 15]]

# 内包表記+zip
Ran_3 = np.array([[random.randint(mi[0], ma[0]), random.randint(mi[1], ma[1])] for mi, ma in zip(Min, Max)])
print(Ran_3)
# [[ 6  7]
#  [ 9  8]
#  [11 13]]

# 関数
def rnd(i, j):
    return random.randint(Min[i, j], Max[i, j])
Ran_4 = np.array([[rnd(i, 0), rnd(i, 1)] for i in range(3)])
print(Ran_4)
# [[ 3  6]
#  [ 7  8]
#  [11 13]]

Comments

[nekohouse]

ありがとうございます！
zip表記など勉強になりました
Ran = np.array([[random.randint(Min[i, j], Max[i, j])for j in range(2)] for i in range(3)])
教えてもらった内包表記を使ってこれで走らせてみたところ大丈夫そうなのでこれでいこうと思います