Changed something for python/numpy compatibility

Travis indicated that passing a function over a set of integers with map was unacceptable for python3.4 and numpy 1.8 so I turned it into a lambda.