2

I need to change my array from the following 

Array = np.array([x1,y1,z1,x2,y2,z2......])

to 

Array = [[x1,x2,x3......]
         [y1,y2,y3,.....]
         [z1,z2,z3,.....]]

Is this possible if so how ?

Thanks

Adam Jaamour
  • 1,326
  • 1
  • 15
  • 31
Gerwyn Jones
  • 33
  • 1
  • 5

3 Answers3

2

You just need to reshape that 1D array to 2D, then transpose it.

import numpy as np

a = np.array([10, 11, 12, 20, 21, 22, 30, 31, 32, 40, 41, 42, 50, 51, 52])
a = a.reshape(-1, 3).T
print(a)

output

[[10 20 30 40 50]
 [11 21 31 41 51]
 [12 22 32 42 52]]
PM 2Ring
  • 54,345
  • 6
  • 82
  • 182
  • You're a star, I never thought about transposing.... I tried using reshape for hours with no luck, this is a much better way than what I did as this can be used for more than three variable (which I might need to do) Thanks :D Is there a way of making this as the answer? – Gerwyn Jones Nov 28 '17 at 23:11
0

I think reshape could help you. See this docs: https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.reshape.html

Yurii Kramarenko
  • 1,025
  • 6
  • 16
0

Is it what you were looking for ?

import numpy as np

foo = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8])
ret = [[], [], []]

for idx, number in enumerate(foo):
    ret[idx % 3].append(number)

print ret # out: [[0, 3, 6], [1, 4, 7], [2, 5, 8]]
IMCoins
  • 3,149
  • 1
  • 10
  • 25
  • 1
    Numpy can do this process _very_ efficiently. Generally, if you're using a Python `for` loop on a Numpy array you're doing it wrong. – PM 2Ring Nov 27 '17 at 12:09