Numpy reshaping array

Question

I have an array reproduced below, when I plot it I get a sawtooth wave, I am looking for a square wave output, ie when the value the 11th value in the second column should decrease in value. I am looking for a way to do this without manually reshaping everytime.

For example I have this:

[[ 0.   0. ]
 [ 0.   0.1]
 [ 0.   0.2]
 [ 0.   0.3]
 [ 0.   0.4]
 [ 0.   0.5]
 [ 0.   0.6]
 [ 0.   0.7]
 [ 0.   0.8]
 [ 0.   0.9]
 [ 0.1  0. ]
 [ 0.1  0.1]
 [ 0.1  0.2]
 [ 0.1  0.3]
 [ 0.1  0.4]
 [ 0.1  0.5]
 [ 0.1  0.6]
 [ 0.1  0.7]
 [ 0.1  0.8]
 [ 0.1  0.9]

I want this:

 [[ 0.   0. ]
 [ 0.   0.1]
 [ 0.   0.2]
 [ 0.   0.3]
 [ 0.   0.4]
 [ 0.   0.5]
 [ 0.   0.6]
 [ 0.   0.7]
 [ 0.   0.8]
 [ 0.   0.9]
 [ 0.1  0.9]
 [ 0.1  0.8]
 [ 0.1  0.7]
 [ 0.1  0.6]
 [ 0.1  0.5]
 [ 0.1  0.4]
 [ 0.1  0.3]
 [ 0.1  0.2]
 [ 0.1  0.1]
 [ 0.1  0.0]

ebarr · Accepted Answer · 2014-04-02 22:58:21Z

1

If the first column is staying the same in the bottom half of the array:

my_array[10:] = my_array[10:][::-1]

Or if your array is not a fixed size:

my_array[my_array.shape[0]/2:] = my_array[my_array.shape[0]/2:][::-1]

answered Apr 2, 2014 at 22:58

ebarr

7,8521 gold badge32 silver badges43 bronze badges

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

Comments

wflynny · Accepted Answer · 2014-04-02 23:11:48Z

1

If you have a specified ymin, ymax, ystep, you could do something like:

import numpy as np
ymin, ymax, ystep = 0, 1, 0.1
z = np.arange(ymin, ymax, ystep)
x = np.repeat(z, len(z))
y = np.tile(np.tile((z, z[::-1]), (1, 1)).flatten(), len(z)/2)
arr = np.vstack((x, y)).T

>>>arr[:20]
array([[ 0. ,  0. ],
       [ 0. ,  0.1],
       [ 0. ,  0.2],
       [ 0. ,  0.3],
       [ 0. ,  0.4],
       [ 0. ,  0.5],
       [ 0. ,  0.6],
       [ 0. ,  0.7],
       [ 0. ,  0.8],
       [ 0. ,  0.9],
       [ 0.1,  0.9],
       [ 0.1,  0.8],
       [ 0.1,  0.7],
       [ 0.1,  0.6],
       [ 0.1,  0.5],
       [ 0.1,  0.4],
       [ 0.1,  0.3],
       [ 0.1,  0.2],
       [ 0.1,  0.1],
       [ 0.1,  0. ]])
       #keeps going
       #...
       #[0.9,  0.9],
       #...
       #[0.9,  0. ]]

edited Apr 2, 2014 at 23:11

answered Apr 2, 2014 at 23:04

wflynny

18.6k6 gold badges50 silver badges69 bronze badges

Collectives™ on Stack Overflow

Numpy reshaping array

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related