Question

I am using this loop in my python code:

final_fun=[]
for i_base in xrange(n_base):
  final_fun.append(sum(fun[:,i_base])/n_ci)

and I would want to know how I can simplify this loop? If I simplify this loop will my program be faster? In general I know how to simplify this but here I am blocked by the append!

Answer 1

Use list comprehensions. It’s faster and cleaner. The interpreter can be slowed down a lot by the for loop, and there’s nothing to lose by simply rearranging the syntax.

See this explanation for more details.

final_fun = [sum(fun[:,i_base])/n_ci for i_base in xrange(n_base)]