convert string to numpy array python

Question

I have a string like that:

'[[-1. ]
 [ 4.5]
 [ 0. ]]

[[ 8.]
 [ 0.]
 [ 6.]]

[[ 0.        ]
 [ 4.        ]
 [ 0.66666667]]'

and I want to convert into a NumPy array like this

array([[[-1.        ],
        [ 4.5       ],
        [ 0.        ]],

       [[ 8.        ],
        [ 0.        ],
        [ 6.        ]],

       [[ 0.        ],
        [ 4.        ],
        [ 0.66666667]]])

i try this code but didn't gat my answer

np.array(list(string.replace(']','],')))

For your firsts steps I will do something like: # Remove all white spaces while string.find(" ") != -1: string = string.replace(" ", "") # Add comma between brackets string = string.replace("][", "],[") Then you will need to convert it into a list object where each element is a string you can convert into an int. — Izaya
– Izaya, Commented Dec 3, 2020 at 11:39
Does this answer your question? Convert 4D array of floats from txt (string) file to numpy array of floats — Daniel F
– Daniel F, Commented Dec 3, 2020 at 11:59

mujjiga · Accepted Answer · 2020-12-03 11:38:22Z

1

If you know the size before hand then

np.fromstring(
    s.replace('[', '').replace(']','').replace('\n', ''), 
                dtype=float, sep=' ').reshape(3,3)

Testcase:

s = '''[[-1. ]
 [ 4.5]
 [ 0. ]]

[[ 8.]
 [ 0.]
 [ 6.]]

[[ 0.        ]
 [ 4.        ]
 [ 0.66666667]]'''

np.fromstring(
    s.replace('[', '').replace(']','').replace('\n', ''), 
                dtype=float, sep=' ').reshape(3,3)

Output:

array([[-1.        ,  4.5       ,  0.        ],
       [ 8.        ,  0.        ,  6.        ],
       [ 0.        ,  4.        ,  0.66666667]])

answered Dec 3, 2020 at 11:38

mujjiga

17.1k2 gold badges37 silver badges54 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

convert string to numpy array python

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related