adding a python string with a numpy array

Question

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?

np.array(line.split(','),int) also works - giving array a list of strings and telling it to interpret them as ints. — hpaulj
– hpaulj, Commented Oct 6, 2022 at 21:43

Mahdi F. · Accepted Answer · 2022-10-06 18:53:00Z

1

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])

answered Oct 6, 2022 at 18:53

Mahdi F.

24.1k5 gold badges25 silver badges32 bronze badges

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

1 Comment

thank you very much, this works well. I was looking at map and where and other functions, and glad to see there is a fromstring() function.