Q&A
素晴らしいです。こんな方法があったとは。
Python3.xで、Numpy配列の内0以外の要素を全て1にする効率の良い方法を教えてください。
Python
1# 返還前 2a = np.array([ 3 [9,0,0,0,9], 4 [9,0,0,1,1], 5 [9,0,8,8,8], 6]) 7 8# ↓変換後のイメージ 9a = np.array([ 10 [1,0,0,0,1], 11 [1,0,0,1,1], 12 [1,0,1,1,1], 13])
回答2件
0
ベストアンサー
元のaの中身を置き換えたいなら、
python
1>>> a[a != 0] = 1 2>>> print(a) 3[[1 0 0 0 1] 4 [1 0 0 1 1] 5 [1 0 1 1 1]]
置き換えずに新しい配列を作るなら、
python
1>>> np.where(a == 0, 0, 1) 2array([[1, 0, 0, 0, 1], 3 [1, 0, 0, 1, 1], 4 [1, 0, 1, 1, 1]])
あるいは、
python
1>>> (a != 0).astype(a.dtype) 2array([[1, 0, 0, 0, 1], 3 [1, 0, 0, 1, 1], 4 [1, 0, 1, 1, 1]])
投稿2021/03/22 13:08
総合スコア24670
0
効率が良いかは分かりませんが。
投稿2021/03/22 12:48
総合スコア979
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。