[英]Numpy Array summing with weights
我有一個二維numpy數組。
每行長三個元素,整數為0-3。 這表示6位整數,每個單元按順序表示兩位。
我正在嘗試將它們轉換為完整的整數。
例如
for i in range(len(myarray)):
myarray[i] = myarray[i][0] * 16 + myarray[i][1] * 4 + myarray[i][2]
例如,我試圖對每一行求和,但是根據[16,4,1]的某個權重向量。
這樣做最優雅的方法是什么? 我想我必須做一些點積然后加一個總和,但我不是百分之百有信心在哪里做點。
點積傾角是正確的,包括您需要的總和。 因此,要獲得目標數組元素和一組權重的乘積之和:
>>> a = np.array([[0,1,2],[2,2,3]])
>>> a
array([[0, 1, 2],
[2, 2, 3]])
>>> weights = np.array([16,4,2])
>>> np.dot(a,weights)
array([ 8, 46])
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.