Per-Class Counts Histogram

Problem

Count how many labels of each class appear in a flat label list.

Return a list of length num_classes where index c is the number of labels equal to c.

case 1

in: [[0,1,1,2,1,0],3]

expected: [2,3,1]

case 2

in: [[0,0,0],2]

expected: [3,0]

case 3

in: [[],3]

expected: [0,0,0]

case 4

in: [[2,2],4]

expected: [0,0,2,0]

Starter codePython

def class_counts(labels, num_classes):
    # Your code here
    pass

4 test cases · run your code against them in the editor