关于计数多维数组唯一数的输出

Question

There is a multi-dimensional array generated from a prediction model. I would like to check the values of this array. Here is my code.

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

This is the result. I do not quite understand the output item marked with yellow. Why it cannot give the exact value instead of 1... . Thanks.

Answer 1

This is just the print output of a numpy array.

You can set the linewidth and edgeitems yourself numpy.set_printoptions

Try this:

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)

Another way would be to convert np.array to list and print that ( docs )

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