[英]How to product element-wise a 2-d numpy array into 3-d over the second dimension of the latter?
我有一個 numpy 矩陣b = np.array([[1,0,1,0],[0,0,0,1]])
並且我想將它按元素生成一個 3-dim 數組a = np.array([[[1,2,3,4], [5,6,7,8], [9,10,11,12]], [[13,14,15,16], [17,18,19,20], [21,22,23,24]]])
用於第二維上的每個索引。 所以,我期望的結果應該如下:
[[[1,0,3,0], [5,0,7,0], [9,0,11,0]], [[0,0,0,16], [0,0,0,20], [0,0,0,24]]]
如果我做a * b
, Numpy 不會廣播。 我正在考慮在其第二維中廣播 b。 我試過np.broadcast_to(b, (2,3,4))
但我得到了錯誤。 我試過(np.broadcast_to(b, (3,2,4)).reshape(2,3,4))
但 output 不如預期。
你需要重塑:
c = b.reshape(2,-1,4)*a
使用None/newaxis
添加一個新的中間維度( reshape
也這樣做):
In [36]: b.shape
Out[36]: (2, 4)
In [37]: a.shape
Out[37]: (2, 3, 4)
In [38]: b[:,None,:]*a
Out[38]:
array([[[ 1, 0, 3, 0],
[ 5, 0, 7, 0],
[ 9, 0, 11, 0]],
[[ 0, 0, 0, 16],
[ 0, 0, 0, 20],
[ 0, 0, 0, 24]]])
In [39]: b[:,None,:].shape
Out[39]: (2, 1, 4)
broadcast_to
無法自動添加該額外維度。 它遵循與b*a
操作相同的規則。 如果需要,它可以添加前導維度,並縮放大小 1 維度。 但是對於其他任何事情,您都必須明確。
In [41]: np.broadcast_to(b, (2,3,4))
Traceback (most recent call last):
File "<ipython-input-41-3c3268de7ce1>", line 1, in <module>
np.broadcast_to(b, (2,3,4))
File "<__array_function__ internals>", line 5, in broadcast_to
File "/usr/local/lib/python3.8/dist-packages/numpy/lib/stride_tricks.py", line 411, in broadcast_to
return _broadcast_to(array, shape, subok=subok, readonly=True)
File "/usr/local/lib/python3.8/dist-packages/numpy/lib/stride_tricks.py", line 348, in _broadcast_to
it = np.nditer(
ValueError: operands could not be broadcast together with remapped shapes [original->remapped]: (2,4) and requested shape (2,3,4)
In [42]: np.broadcast_to(b[:,None,:], (2,3,4))
Out[42]:
array([[[1, 0, 1, 0],
[1, 0, 1, 0],
[1, 0, 1, 0]],
[[0, 0, 0, 1],
[0, 0, 0, 1],
[0, 0, 0, 1]]])
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.