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, Jun 15 '18 at 19:58
@PaulPanzer s = '[ 0. 0. 1.]' np.matrix(s) returns 'TypeError: Invalid data string supplied: [' — blue-sky, Jun 15 '18 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, Jun 15 '18 at 20:54

score 6 · Accepted Answer · answered Jun 15 '18 at 20:20

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 '18 at 20:20

anishtain4

2,042
1
16
19

score 0 · Answer 2 · answered Jun 15 '18 at 19:59

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 '18 at 19:59

rahlf23

8,613
3
19
49

Convert string array to numpy array

2 Answers2

Linked