Brad Spendlove - 7 months ago 37

Python Question

I'm following this tutorial on how to use numpy to manipulate images. When I load the sample image using scipy, I get a 2D array of RGB tuples, with a dtype value appended on the end.

`array([[7, 8, 5],`

[3, 5, 7]], dtype=uint8)

I wrote a function and vectorized it

`def myfunc(a, b):`

return a + 2

vfunc = np.vectorize(myfunc)

but when I apply it to my array, the result doesn't have the dtype

`array([[9, 10, 7],`

[5, 7, 9]])

My guess is that because "dtype + 2" isn't defined, it's just losing that element of the array.

How can I write a function that will not strip the dtype when I vectorize it and apply it to a numpy array?

Answer

`dtype=uint8`

is not an element of the array. It is just a thing that gets printed to let you know that the array is of type `np.uint8`

.

The default types `np.float_`

and `np.int_`

do not get a printout like that, which is what you are seeing in the second case. The way you can tell float and int arrays apart is that float arrays will always have decimal points in the numbers.

The reason that this is happening is that you are adding `2`

to each element of your array. Since `2`

is an integer, the output array gets promoted to `np.int_`

type and you do not get an explicit `dtype`

printout.

You can try the following experiment: redefine `myfunc`

to add a `np.uint8`

instead of an integer to the array elements and try to print the result:

```
def myfunc(a, b):
return a + np.uint8(2)
```

Finally, keep in mind that vectorizing Python code is usually not the best way to get things done. The function itself will be a Python function, and therefore slow. It is generally better to find a way of performing whatever operations you want with numpy functions.