Разложение в ряд на питоне

x = float(input())
N = int(input())
e = float(input())
a0 = 1 
s = a0
s2 = 0
k = 0 
#1 случай
for i in range(N):
    k +=1
    a1 = (-x)**k
    s = s + a1 
    if abs(a1) > e:
        s2 += a1
    a0 = a1
print(s)
print(s2)
#2 случай

x = 0.3 #float(input())
n = 100 #int(input())
e = 0.000001 #float(input())
s = 0
s2 = 0
#1 случай
for i in range(n):
    a1 = (-x)**i
    s = s + a1
    if abs(a1) > e:
        s2 += a1
print(s)
print(s2)
print(1 / (1 + x))
 
#2 случай
def foo(x, e):
    i = 0
    p = 1
    s = 0
    while abs(p) > e:
        p = (-x) ** i
        s += p
        i += 1
    return i, s
print(foo(x, 0.000001))
print(foo(x, 0.00001))
print(1 / (1 + x))

x = .9 #float(input())
N = 50 #int(input())
e = 1e-2 #float(input())
a0 = 1 
s = s2 = k = 0 
#1 случай
for i in range(N):
    s += a0
    if abs(a0) > e:
        s2 += a0
        k +=1
    a0 *= -x
 
print(round(s, 7))
print(round(s2, 7), k)
print(round(1/(1+x), 7))
print()
 
#2 случай
e1 = 1e-5
e2 = e1/10
s1 = s2 = k1 = k2 = 0
a0 = 1
 
while abs(a0) > min(e1, e2):
    if abs(a0) > max(e1, e2):
        s1 += a0
        k1 += 1
        s2, k2 = s1, k1
    else:
        s2 += a0
        k2 += 1
    a0 *= -x
 
print(round(s1, 7), k1)
print(round(s2, 7), k2)