CODEWITHSUNDEEP
pythonarrayslistsplit
zeb92
zeb92

Reputation: 83

Split each string in list and store in multiple arrays

I have a list that looks like this: ['1 0 0','2 0.5 0.25','3 1 0.5','4 1.5 0.75']

I want to end up with these three arrays: [1,2,3,4] and [0,0.5,1,1.5] and [0,0.25,0.5,0.75]

i.e. I want the first value of each list item and store it in an array, and do the same with the second and third values.

I tried this

for i in coordinates[:]:
    number,x,y=i.split(' ')

also tried using number[] and number.append but none of these seem to work

Upvotes: 0

Views: 3773

Answers (4)

Aaditya Ura
Aaditya Ura

Reputation: 12669

Here is one approach without for loop , using lambda :

a=['1 0 0','2 0.5 0.25','3 1 0.5','4 1.5 0.75']

print(list(zip(*list((map(lambda y:y.split(),a))))))

output:

[('1', '2', '3', '4'), ('0', '0.5', '1', '1.5'), ('0', '0.25', '0.5', '0.75')]

Upvotes: 0

Manuel
Manuel

Reputation: 706

Creating three new list, and going through a input:

input = ['1 0 0','2 0.5 0.25','3 1 0.5','4 1.5 0.75']
x_list = []
y_list = []
z_list = []

for i in input:
    x, y, z = i.split(' ')
    x_list.append(x)
    y_list.append(y)
    z_list.append(z)

Upvotes: 1

Paul Panzer
Paul Panzer

Reputation: 53029

Similar to @ikkuh's but with "configurable" types:

data = ['1 0 0','2 0.5 0.25','3 1 0.5','4 1.5 0.75']
tp = [int, float, float] 

parsed = ((t(j) for t, j in zip(tp, record.split())) for record in data)
idx, x, y = (list(i) for i in zip(*parsed))
idx
# [1, 2, 3, 4]
x
# [0.0, 0.5, 1.0, 1.5]
y
# [0.0, 0.25, 0.5, 0.75]

Upvotes: 2

ikkuh
ikkuh

Reputation: 4603

This can be done as follows:

list(zip(*(list(map(float, s.split())) for s in coordinates)))

First we loop through all strings in the list and split them

[s.split() for s in coordinates]

Then we map the float function over all the individual strings to convert them to floats:

[list(map(float, s.split())) for s in coordinates]

Then we use zip to get them the way you want them.

Upvotes: 6

Related Questions