numpy的數組：價值分配失敗

Question

我有這樣的代碼片段：

import numpy as np 

a=np.array([5,6,7,8,9]) 
b=np.array([5,6,7,8,9]) 

scoreA = np.array([float(1)/(i + 1) for i in range(len(a))]) 
scoreB = np.array([0 for i in range(len(b))]) 

for eleA in a: 
    if eleA in b: 
     i, = np.where(b == eleA) 
     i = i[0] 
     j, = np.where(a == eleA) 
     j = j[0] 
     scoreB[i] = scoreA[j] 

     print "B is: %f" % scoreB[i] 
     print "A is: %f" % scoreA[j] 

所以其基本思路是：對數組a和b，如果一個元素在他們兩人發現，然後我會分配scoreA中的元素得分爲B。但結果是這樣的：

B is: 1.000000 
A is: 1.000000 
B is: 0.000000 
A is: 0.500000 
B is: 0.000000 
A is: 0.333333 
B is: 0.000000 
A is: 0.250000 
B is: 0.000000 
A is: 0.200000 

這意味着該行：

scoreB[i] = scoreA[j] 

工作不正常？我該如何解決這個問題？

Answer 1

你的解決方案很奇怪，最好把它作爲@Divakar提到。 但是無論如何，直接的問題是scoreA和scoreB有不同的類型：float64和int64。

使

scoreA = np.array([float(1)/(i + 1) for i in range(len(a))],dtype=float) 
scoreB = np.array([0 for i in range(len(b))],dtype=float) 

或其他一些D型，以確保您所有的成績都相同的類型。