Home > Software engineering >  adding a python string with a numpy array
adding a python string with a numpy array

Time:10-07

I have a string of numbers like this:

line
Out[2]: '1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0'

I have a numpy array of numbers like this:

rpm
Out[3]:
array([1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
       0., 0., 0., 0., 0., 0., 0., 0.])

I'd like to add the line of numbers in the string to the numbers in the array, and keep the result in the array.

Is there a simple/elegant way to do this using numpy or python?

CodePudding user response:

You can use numpy.fromstring.

import numpy as np

line = '1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0'
res = np.fromstring(line, dtype=int, sep=',')
print(res)

Output:

array([1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
       0, 0])
  • Related