fork download
  1. f = lambda x,e=enumerate:[sum(t/1j**(4*k*n/len(x))for n,t in e(x))for k,_ in e(x)]
  2.  
  3. for x in ([1, 1, 1, 1], [1, 0, 2, 0, 3, 0, 4, 0], [1, 2, 3, 4, 5], [5-3.28571j, -0.816474-0.837162j, 0.523306-0.303902j, 0.806172-3.69346j, -4.41953+2.59494j, -0.360252+2.59411j, 1.26678+2.93119j]):
  4. print(f(x), '\n')
Success #stdin #stdout 0.02s 9984KB
stdin
Standard input is empty
stdout
[(4+0j), 0j, 0j, 0j] 

[(10+0j), (-2+2j), (-2+0j), (-2-2j), (10+0j), (-2+2j), (-2+0j), (-2-2j)] 

[(15+0j), (-2.5+3.440954801177933j), (-2.5000000000000004+0.8122992405822647j), (-2.499999999999999-0.8122992405822678j), (-2.4999999999999956-3.440954801177935j)] 

[(2.0000019999999994+5.9999999995064e-06j), (-2.1873894540114236e-06-2.9999907158676136j), (4.999996876255548+5.070678220420177e-06j), (-7.704864037894055e-06-6.999998252517274j), (11.000004936866294+2.3649529490299415e-07j), (1.217868931480126e-06-12.99999945557741j), (17.00000486126272+7.116788791061701e-06j)]