Extract first n characters from each element of a numpy string array

Question

I have a NumPy array which elements are strings. For example

import numpy as np    
a = np.array(["abcde", "degag"])

I need to extract the first n characters from each element of the array (substr/strleft). The result for the first 3 characters should be an aray like this one:

array(['abc', 'deg'], dtype='<U3')

Thank you very much

hpaulj · Accepted Answer · 2021-08-10 20:39:01Z

5

It's almost staring you in the face:

In [93]: a = np.array(["abcde", "degag"])
In [94]: a.astype('U3')
Out[94]: array(['abc', 'deg'], dtype='<U3')

otherwise you have to iterate

In [95]: [s[:3] for s in a]
Out[95]: ['abc', 'deg']

answered Aug 10, 2021 at 20:39

hpaulj

233k14 gold badges260 silver badges392 bronze badges

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

Collectives™ on Stack Overflow

Extract first n characters from each element of a numpy string array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related