Ускорение кода Python на 2 секунды (исходные 2,003 с)

Содержимое статьи:

• Проблема
• Решение
• Оптимизированный код

Проблема

Исходный код решает задачу, связанную с интервалами, и выполняется медленно.

Решение

1. Инициализация массива: Изначально массив x инициализируется нулями. Использование memset ускоряет эту операцию.
2. Подсчет неперекрывающихся интервалов: Перебор массива t и увеличение соответствующих элементов в x можно ускорить с помощью numpy.
3. Подсчет элементов, равных нулю: Подсчет нулей в массиве x можно ускорить с помощью numpy.
   

   Оптимизированный код

   import numpy as np
   
   import time
   
   k, n = 100000, 300
   
   t = []
   
   for _ in range(k):
   
   x1, x2 = np.random.randint(0, 479, size=2)
   
   t.append((np.min(x1, x2), np.max(x1, x2)))
   
   start_time = time.time()
   
   x = np.zeros([480], dtype=np.int32)
   
   for i in t:
   
   x[i[0] : i[1] + 1] += 1
   
   print(n - np.count_nonzero(x))