Convert strings within an array to list in Python

Question

I want to convert the array A containing string elements to list. But I am running into error while using split(). I present the expected output.

import numpy as np
from numpy import nan
A=np.array(['[1]', '[2]',  '[3]', '[4]', '[5]'])
A.split()
print(A)

The error is

in <module>
    A.split()

AttributeError: 'list' object has no attribute 'split'

The expected output is

array([[1], [2], [3], [4], [5]])

Try this: A = np.array([eval(e) for e in A])

Dogan Askan
– Dogan Askan

2022-09-11 16:20:43 +00:00
Commented Sep 11, 2022 at 16:20 — Dogan Askan
– Dogan Askan, Commented Sep 11, 2022 at 16:20
A string has a split method, not a list!

hpaulj
– hpaulj

2022-09-11 16:20:46 +00:00
Commented Sep 11, 2022 at 16:20 — hpaulj
– hpaulj, Commented Sep 11, 2022 at 16:20

Lucas M. Uriarte · Accepted Answer · 2022-09-11 16:25:50Z

2

I think this should work

A=np.array(['[1]', '[2]',  '[3]', '[4]', '[5]'])
output = [[int(a.strip("[|]"))] for a in A]
print(output)

[[1], [2], [3], [4], [5]]

if you prefer just an list, instead of a list of list

A=np.array(['[1]', '[2]',  '[3]', '[4]', '[5]'])
output = [int(a.strip("[|]")) for a in A]
print(output)

[1, 2, 3, 4, 5]

edited Sep 11, 2022 at 16:25

answered Sep 11, 2022 at 16:16

Lucas M. Uriarte

3,17611 silver badges25 bronze badges

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

3 Comments

user19862793 Over a year ago

The output I am getting with your code is [array(['[1]', '[2]', '[3]', '[4]', '[5]'], dtype='<U3')].

Lucas M. Uriarte Over a year ago

I have edited the the code if you print output is impossible you are getting this. Things apart I don't get why you are using numpy to have an array of strings.

Lucas M. Uriarte Over a year ago

I think you are printing A and not output. A is the array you have as input, output is what you want

uozcan12 · Accepted Answer · 2022-09-11 17:08:39Z

0

import numpy as np
A = np.array(['[1]', '[2]',  '[3]', '[4]', '[5]'])
B = np.array([int(item.strip("'\'").replace("[","").replace("]","")) for item in A] )
print(B) #[1 2 3 4 5]

answered Sep 11, 2022 at 17:08

uozcan12

4251 gold badge5 silver badges13 bronze badges

Collectives™ on Stack Overflow

Convert strings within an array to list in Python

2 Answers 2

3 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

3 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related