Convert string array to numpy array

Question

I'm attempting to convert the string '[ 0. 0. 1.]' to a numpy array.

This is the code I've written but is more complicated that needs be ?

arr = []
s = '[ 0.  0.  1.]'
arr.append(int(s.split(" ")[1].replace("." , '')))
arr.append(int(s.split(" ")[3].replace("." , '')))
arr.append(int(s.split(" ")[5].replace("]" , '').replace("." , '')))

arr = np.array(arr)

print(arr)
print(type(arr))
print(type(arr[0]))

Above code prints :

[0 0 1]
<class 'numpy.ndarray'>
<class 'numpy.int64'>

Is there a cleaner method to convert string '[ 0. 0. 1.]' to numpy int array type ?

depends on how flexible you want to be. np.array(s[1:-1].split(), float).astype(int) — Paul Panzer
– Paul Panzer, Commented Jun 15, 2018 at 19:58
@PaulPanzer s = '[ 0. 0. 1.]' np.matrix(s) returns 'TypeError: Invalid data string supplied: [' — blue-sky
– blue-sky, Commented Jun 15, 2018 at 20:06
Looks like it changed sometime during 1.13.x. So if you are 1.12 or less you'll have to do np.matrix(s[1:-1]).A1.astype(int) — Paul Panzer
– Paul Panzer, Commented Jun 15, 2018 at 20:54

anishtain4 · Accepted Answer · 2018-06-15 20:20:28Z

6

Numpy as can handle it much easier than all the answers:

s = '[ 0.  0.  1.]'
np.fromstring(s[1:-1],sep=' ').astype(int)

answered Jun 15, 2018 at 20:20

anishtain4

2,4102 gold badges18 silver badges22 bronze badges

Add a comment |

rahlf23 · Accepted Answer · 2018-06-15 19:59:22Z

0

IN:

import numpy as np

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

out = np.array([int(i.replace('.','')) for i in s[s.find('[')+1:s.find(']')].split()])

print(type(out))

OUT:

<class 'numpy.ndarray'>

answered Jun 15, 2018 at 19:59

rahlf23

9,0394 gold badges30 silver badges57 bronze badges

Add a comment |

Collectives™ on Stack Overflow

Convert string array to numpy array

2 Answers 2

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Your Answer

Sign up or log in

Post as a guest

Linked

Related