linebyline.py

def profile(func):
    from functools import wraps

    @wraps(func)
    def wrapper(*args, **kwargs):
        from line_profiler import LineProfiler
        prof = LineProfiler()
        try:
            return prof(func)(*args, **kwargs)
        finally:
            prof.print_stats()

    return wrapper
  
 
@profile
def square_my_num():
    universe = np.arange(100000)
    numbers = np.random.rand(100000)
    
    filtered_nums=[]
    
    for number in numbers:
        if number in universe:
            filtered_nums.append(number)

    squares = []
    for number in filtered_nums:
        squares.append(number ** 2)
@profile
def matrix_stuff():
    a = np.random.rand(5000,5000)
    for i in range(a.shape[0]):
        for j in range(a.shape[1]):
            if a[i][j] < 0:
                a[i][j] = a[i][j]*0
            else:
                a[i][j] = a[i][j]*1

def main():
    matrix_stuff()
    square_my_num()
    # square_my_num_faster()