#!/usr/bin/env python3

import numpy

def numpysum(n):
    a = numpy.arange(n) ** 2
    b = numpy.arange(n) ** 3
    c = a + b
    return c

print(numpysum(3)) # [ 0  2 12]