In Python, a list comprehension is a convenient way to create a list.
def f(x):
return x**2
x = [1,2,3,4,5]
#------------ Method 1
y = []
for i in range(len(x)):
y.append(f(x[i]))
#------------ Method 2: list comprehension
y = [ f(x[i]) for i in range(len(x)) ]
import numpy as np
x = np.array([1,2,3,4,5])
#------------ Method 1
y = np.zeros(len(x))
for i in range(len(x)):
y[i] = f(x[i])
#------------ Method 2: list comprehension
y = np.array([ f(x[i]) for i in range(len(x)) ])
x = [5 for i in range(6)]
x
[5, 5, 5, 5, 5, 5]
x = [i for i in range(6)]
x
[0, 1, 2, 3, 4, 5]
x = [i**2 for i in range(6)]
x
[0, 1, 4, 9, 16, 25]
x = [i for i in range(10) if i%2==0] # same as x=[i for i in range(0,10,2)]
x # but illustrates using an "if" condition
[0, 2, 4, 6, 8]