pythonのnumpyで行列の特定の要素を０にしたい

Question

pythonのnumpyで生成した行列の特定の要素を０にしたいのですが、いまいち上手な方法が思い浮かびません。
具体的には0~1の乱数で生成した行列の0.5以下を0にしたいと考えています。

Answer 1

これでよいのではないでしょうか。

import numpy as np

# 0〜1の乱数にて 5x5 の行列を生成
d = np.random.rand(5,5)

# 0.5以下を0にする
d[d<=0.5] = 0

print(d)

Comments

[simasima]

ありがとうございます！とてもシンプルなので、ベストアンサーにしました！！

Answer 2

1行で出来るcip関数で行けるかと思ったのですが、ダメでしたo.0
どうしてもフラグ処理が必要っぽいです。

# coding: UTF-8
import numpy as np

array = np.arange(10)

print(array)

lim = 5
flag = array < lim
array[flag] = 0

print(array)

# Upper threshold
# clip_max = 1.0
# 
# # Lower threshold
# clip_min = 0.5
#
# print(np.clip(array,clip_min,clip_max))

Comments

[simasima]

解答ありがとうございます！配列処理はなかなか難しいですね。。。