Cluster breakdown in Week 3 live HW

gistfile1.txt

clusters = {}

n = len(column_values)

for i in range(n):
    if column_values[i] not in clusters:
        clusters[column_values[i]] = {'count': 0, 'purch': 0}

    clusters[column_values[i]]['count'] += 1

    if y[i] == 1:
        clusters[column_values[i]]['purch'] += 1

def make_strings(c, n):
    for i in range(len(c)):
        count = c[i]['count']
        purch = c[i]['purch']

        c[i]['r_str'] = f"{i}: {100 * count / n :0.2f}%"
        c[i]['p_str'] = f"{purch :5d} out of {count} -- {100 * purch / count :0.2f}%"

    return c


def print_string(clusters, string_type):
    for i in range(len(clusters)):
        print(f"\t{clusters[i][string_type]}")


clusters = make_strings(clusters, n)

print("\nRatios\n")
print_string(clusters, 'r_str')

print("\nPurchases\n")
print_string(clusters, 'p_str')