[FIXED] How to convert string formed by numpy.array2string back to array?

Issue

I have a string of numpy array which is converted by using numpy.array2string

Now, I want back my numpy array.

Any suggestions for how I can achieve it?

My Code:

img = Image.open('test.png')
array = np.array(img)
print(array.shape)
array_string = np.array2string(array, precision=2, separator=',',suppress_small=True)

P.S My array is a 3D array not 1D and I am using , separators, not the default blank

Solution

Update: I just tried this and it worked for me:

import numpy as np
from PIL import Image

img = Image.open('2.jpg')

arr = np.array(img)

# get shape and type
array_shape = arr.shape
array_data_type = arr.dtype.name
# converting to string
array_string = arr.tostring()

# converting back to numpy array
new_arr = np.frombuffer(array_string, dtype=array_data_type).reshape(array_shape)
print(new_arr)

For converting numpy array to string, I used arr.tostring() instead of arr.array2string(). After that converting back to numpy array works with np.frombuffer().

Answered By - Scott