map int over 2D array python

Question

To parse

s="1,2,3,4_5,6,7,8"

as [[1,2,3,4],[5,6,7,8]]

I am currently using

import numpy as np
a=np.array([list(map(int,r.split(","))) for r in s.split("_")])

Is there a more pythonic or one-shot inbuilt way of doing this or am I on the right track here?
Python newbie.

rdas · Accepted Answer · 2020-02-17 14:00:26Z

2

Using list-comprehensions:

s="1,2,3,4_5,6,7,8"

a = np.array([[int(x) for x in r.split(',')] for r in s.split('_')])

answered Feb 17, 2020 at 14:00

rdas

21.4k6 gold badges39 silver badges48 bronze badges

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

Comments

sacuL · Accepted Answer · 2020-02-17 14:01:11Z

2

You can use np.genfromtxt:

from io import StringIO
import numpy as np

s="1,2,3,4_5,6,7,8"

np.genfromtxt(StringIO(s.replace("_", "\n")), delimiter=",")

array([[1., 2., 3., 4.],
       [5., 6., 7., 8.]])

answered Feb 17, 2020 at 14:01

sacuL

51.6k9 gold badges88 silver badges115 bronze badges

2 Comments

yatu Over a year ago

Ah nice answer! Why is StringIO necessary here?

chepner Over a year ago

genfromtxt reads from a file-like object, which StringIO provides.

Collectives™ on Stack Overflow

map int over 2D array python

2 Answers 2

Comments

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related