Skip to content

Commit bb7e804

Browse files
committed
add Ramanujan sum
1 parent aa1b315 commit bb7e804

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

ramanujan_sum.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
# Author Dario Clavijo 2023
2+
# based on https://en.wikipedia.org/wiki/Ramanujan%27s_sum
3+
4+
from gmpy2 import *
5+
6+
ipi2 = (2j) * complex(gmpy2.const_pi())
7+
8+
def c(q,n):
9+
sum_ = 0
10+
nipi2 = ipi2 * n
11+
for a in range(1, q+1):
12+
if gcd(a,q) == 1:
13+
sum_ += exp(nipi2 * (a/q))
14+
return sum_
15+
16+
print([c(1,n).real for n in range(1, 31)])

0 commit comments

Comments
 (0)