len()を利用した要素数のカウントについて

Question

dice_data = np.array([1, 2, 3, 4, 5, 6])
count_all_dice = random.choice(dice_data, 100)
for i in dice_data:
    print(i, "が出る確率", len(count_all_dice[count_all_dice==i]) / calc_steps)

上記のようにサイコロを想定し、それぞれの確率を求めるプログラムです。
このコードの最後の行で、len()を利用してcout_all_diceのそれぞれの要素数を計算しているようですが、なぜこの様なことができるか分かりません。len(count_all_dice)で要素数が計算できることは分かるのです、その後のコードは何らかの条件式でしょうか？
また、count()などは利用できないのでしょうか？
初心者丸出しですが、ご回答よろしくお願い致します。

Answer 1

例えば、次のようなNumPyアレイがあるとします。

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6])

NumPyはブロードキャスト演算に対応しているので、条件式をアレイに適用することが出来ます。

result = (arr==6)    # 6と等しい要素
print(result)        # [False False False False False  True]

result = (arr%2==0)  # 2の剰余が0である要素 (つまり偶数)
print(result)        # [False  True False  True False  True]

---

さらに、NumPyにはindexingという機能があります。
例えば次のような感じです。

arr2 = arr[np.array([True, False, True, True, False, True])]   # 適当に決める
print(arr2)   # [1 3 4 6] Trueのところだけ切り出される

---

上記二つを組み合わせると、次のような記述が出来ます。

arr3 = arr[arr%2==0]   # 偶数の要素だけ切り出す
print(arr3)            # [2 4 6]

結論

len(count_all_dice[count_all_dice==i])は、
count_all_diceのうちiと等しい要素だけ切り出して、その個数を数えています。

Comments

[KojiroYamamoto]

true,falseの数をカウントしていたのですね！
ありがとうございます！

[LouiS0616]

正確に言えば、条件式がtrueである要素だけ切り出して、その長さを求めています。
結果的にはtrueの個数を数えているのと一緒にはなりますが。

Answer 2

count_all_dice[count_all_dice==i]で要素がiに一致するところだけを切り出してきています。試しに==の前後だけを切り出すと分かると思います。bool型の配列を使ったindexの指定ですね。

Comments

[KojiroYamamoto]

bool型なんですね！
ありがとうございます！

[tachikoma]

ですね。なんで、lenをとらずにnp.sum(count_all_dice==i)でも同じ結果が得られますよ。理由はtrueが1, falseが0として処理されるためです。

Answer 3

条件式で、false なら０、trueなら１となるってことですね

つまり、
len(count_all_dice[0])
あるいは
len(count_all_dice[1])
を取っているんです

Comments

[y_waiwai]

ピント外れな回答でしたね。マイナスで沈めておいてくださいませ

[KojiroYamamoto]

いえ、ご回答ありがとうございます！