Посчитать кол-во чисел со всевозможными кол-вами делителей - Python - Ответ 15305242

from collections import Counter
from math import log
import numpy as np
import matplotlib.pyplot as plt
 
 
def primes(n):
    nn = int(np.sqrt(n)) + 1
    pr = np.ones(nn, dtype='int')
    for p in range(3, int(np.sqrt(nn)) + 1, 2):
        if pr[p]:
            pr[np.arange(p * p, nn, 2 * p)] = 0
    res = np.array(np.nonzero(pr)).ravel()
    res = res[(res % 2 == 1)]
    res[0] = 2
    return res
 
 
def main():
    k = int(input())
    pr = primes(k)
    arr = np.arange(1, k + 1)
    res = np.ones(k, dtype=int, order='F')
 
    for p in pr:
        tmp = np.ones(k, dtype=int, order='F')
        npow = int(log(k, p)) + 1
 
        for i in range(1, npow):
            tmp[(arr % p == 0)] += 1
            arr[(arr % p == 0)] //= p
 
        res = np.multiply(tmp, res)
 
    arr[(arr > 1)] = 2
    res = np.multiply(arr, res)
    res = Counter(res.tolist())
    solution = sorted(res.items())
    for group_ in solution:
        print(f"Количество чисел с {group_[0]} делителем: {group_[1]}")
 
    fig, ax = plt.subplots()
    ax.set_facecolor('seashell')
    fig.set_facecolor('floralwhite')
    fig.set_figwidth(12)
    fig.set_figheight(6)
    ax.set_ylabel("Количество чисел")
    ax.set_title("Количество делителей")
    x_s = [group_[0] for group_ in solution]
    # plt.xticks(np.arange(0, max(x_s) + 1, 5))
    y_s = [group_[1] for group_ in solution]
    # plt.yticks(np.arange(0, max(y_s) + 1, 10000))
 
    for group_ in solution:
        x = group_[0]
        y = group_[1]
        ax.bar(x, y)
 
    plt.show()
 
 
main()