關於計數多維數組的唯一編號的輸出

Question

存在從預測模型生成的多維數組。我想檢查這個數組的值。這是我的代碼。

print('Predictions shape ',predictions.shape)   
    unique_items, counts = np.unique(predictions,return_counts=True)   
    print('unique_items ',unique_items) 
    print('counts ',counts) 

這就是結果。我不太瞭解標有黃色的輸出項目。爲什麼它不能給出確切的值而不是1...。謝謝。

Answer 1

這僅僅是一個numpy的陣列的打印輸出。

您可以設置linewidth和edgeitems自己numpy.set_printoptions

試試這個：

import numpy as np 

np.set_printoptions(linewidth=1000, edgeitems=5) 

predictions = np.random.rand(3249, 4096, 2) 

print('Predictions shape ',predictions.shape)   
unique_items, counts = np.unique(predictions,return_counts=True)   
print('unique_items ',unique_items) 
print('counts ',counts) 

另一種方法是將np.array轉換爲list並打印（docs）

# beware, this takes like forever :) 
print('unique_items ',unique_items.tolist()) 
print('counts ',counts.tolist())