作者:采石工
鏈接:https://www.zhihu.com/question/49571479/answer/116840869
來源:知乎
著作權歸作者所有嫌变。商業(yè)轉載請聯(lián)系作者獲得授權沙峻,非商業(yè)轉載請注明出處。
在NumPy中可以利用如下的函數(shù)進行數(shù)組或矩陣的連接:
hstack : Stack arrays in sequence horizontally (column wise).
vstack : Stack arrays in sequence vertically (row wise).
dstack : Stack arrays in sequence depth wise (along third axis).
concatenate : Join a sequence of arrays together.
r_ : Translates slice objects to concatenation along the first axis.
c_ : Translates slice objects to concatenation along the second axis.
題主需要的是在第三維(或軸)上的連接屎即,見如下的代碼:
<pre><code>
#coding=utf-8importnumpyasnp# 因為是生成隨機數(shù)做測試,設置固定隨機數(shù)種子,可以保證每次結果一致??
np.random.seed(0)
RGB=np.random.randint(0,255,(5,5,3))
alpha=np.random.randint(0,255,(5,5))
RGBA=np.dstack((RGB,alpha))
print('RGB = \n {}'.format(RGB))
print('alpha = \n {}'.format(alpha))
print('RGBA = \n {}'.format(RGBA))
print('RGBA[:, :, 3] = \n {}'.format(RGBA[:,:,3]))#coding=utf-8importnumpyasnp# 因為是生成隨機數(shù)做測試乾吻,設置固定隨機數(shù)種子灵疮,可以保證每次結果一致??
np.random.seed(0)
RGB=np.random.randint(0,255,(5,5,3))
alpha=np.random.randint(0,255,(5,5))
RGBA=np.dstack((RGB,alpha))
print('RGB = \n {}'.format(RGB))
print('alpha = \n {}'.format(alpha))
print('RGBA = \n {}'.format(RGBA))
print('RGBA[:, :, 3] = \n {}'.format(RGBA[:,:,3]))
</code></pre>
這是一個代碼區(qū)塊织阅。