With numpy optimization variant