Геометрическая прогрессия

[Nina2016]

Добрый день. Помогите написать программу: Написать все элементы массива цифры которого образуют геометрическую прогрессию. Написала программу определения образуют ли числа массива геометрическую прогрессию, а данная задача никак не получается.

from random import randint
n=4
a=[0]*n
for i in range(n):
    a[i]=float(input())
print(a)
b=[]
k=0
q=a[1]/a[0]
for i in range(1,len(a)):
   if a[i]/a[i-1]==q:
      k=k+1
if k==n-1:
    print('геометрическая')
else:
    print('не геометрическая')