prime_diff.py

import matplotlib.pyplot as plt

primes = []
for i in range(2, 10000):
    for j in range(2, i):
        if i % j == 0:
            break
    else:
        primes.append(i)

diffs = []
for n in primes:
    i = primes.index(n)
    if not i + 1 >= len(primes):
        diffs.append(primes[i + 1] - n)

summ = 0
for n in diffs:
    summ += n

summ /= len(diffs)
print(summ)

diffs.append(0)
plt.plot(primes, diffs)
plt.show()