import numpy as np
import matplotlib.pyplot as plt


a = np.array([1, 2, 3])
print(a)

plt.plot([1, 2, 3], [2, 4, 6])
plt.show()

for num in [1, 2, 3, 4]:
    print(num)


def sqaure(x):

    return x**2
