作者sma1033 (死马)
看板Python
标题Re: [问题] numpy broadcasting 的使用
时间Thu Jun 14 00:24:38 2018
※ 引述《NMOSFET (NMOSFET)》之铭言：
: 各位大大好，小的最近遇到一个问题，
: 假设有个 numpy 二维矩阵 A，
: 我想要让 A 矩阵里的值，
: 以 10% 的机率设为 0，
: 於是写了以下程式码 :
: rows = 10
: cols = 10
: prb = np.random.random((rows, cols))
: mask = prb<0.1
: for row in range(rows):
: for col in range(cols):
: if(mask[row][col]):
: A[row][col]=0
: 有没有能够利用numpy的broadcasting来达到同样的效果，
: 而不是用 for 回圈，因为上面这种写法，
: 我只要在if()里面产生随机数判断就好，
: 不用在上面宣告 prb 和 mask 来多此一举，
: 能否请各位大大帮我解惑，感恩!!
用for回圈在这边根本脱裤子放屁
如果是我的话，会用以下作法
mask = np.random.random((row, col))
mask[prb<0.1] = 0
mask[prb>=0.1] = 1
A = np.multiply(A, mask)
done.
速度要快就是要用矩阵同时处理多element（可以平行算）
如果你用for回圈一个一个element依序处理A当中的资料
不管你用什麽方法生成random都一样会慢
--
※ 发信站: 批踢踢实业坊(ptt.cc), 来自: 61.231.188.20
※ 文章网址: https://webptt.com/cn.aspx?n=bbs/Python/M.1528907083.A.5B5.html
1F：→ sma1033: 看有没有人知道mask 3元运算值的写法，可以再少一行 XD 06/14 00:36
※ 编辑: sma1033 (61.231.188.20), 06/14/2018 00:45:05
2F：推 bibo9901: 应该不用multiply, 直接 A[mask<0.1] = 0 就可以了 06/14 02:39
3F：→ bibo9901: A[np.random.random((row, col)) < 0.1] = 0 06/14 02:39
4F：→ sma1033: A[np.random.random((A.shape[0],A.shape[1]))<0.1]=0 06/14 03:27
5F：→ sma1033: bibo的方法应该是最简洁的方式，推一个 06/14 03:27
6F：→ bibo9901: 要这样的话直接 A[np.random.random(A.shape)<0.1]=0 06/14 08:39
7F：推 NMOSFET: 太感谢了! 谢谢楼上两位大大的帮忙! 06/14 10:59
8F：推 tn00364361: b大的index是不是该改成`prb<0.1`而不是`mask<0.1`? 06/14 12:11
9F：→ tsoahans: np.random.binomial(1, 0.9, (row,col)) 06/14 16:01