Python：如何以最簡單的方式比較兩個數組

Question

我正在做機器學習的東西。 我今天遇到了一個問題，希望有人能幫助我。

我有兩個數組[數組（預測）和數組（標籤））具有相同的尺寸[形狀：（128,5）]。第一個維度是每個預測和標籤的索引。第二維是與每個預測和標籤相關的結果

我想輸出精度 這是我的代碼。

right_count = 0 
for i in range(Prediction.shape[0]) # Foreach each predict/label 
    if（np.array_equal(Prediction[i], Labels[i])): # Compare each result 
     right_count += 1 

accuracy = float(right_count)/Prediction.shape[0]' 

我只是想知道是否有更好的方法來簡單的下面的代碼。

謝謝

Answer 1

該代碼可以簡化，例如：

accuracy = (Prediction == Labels).all(axis=1).mean()

這也將運行比原來的版本速度更快，因爲更多的操作都是內部numpy的，而不是在Python完成碼。