set values of 2D numpy array by parameters

Question

I need to create 2D np array by the known shape and parameters in this way:

import numpy as np
rows = 9
cols = 8
xrows = 3
xcols = 2

the wanted results:

([[ 1,  1,  4,  4,  7,  7, 10, 10],
    [ 1,  1,  4,  4,  7,  7, 10, 10],
    [ 1,  1,  4,  4,  7,  7, 10, 10],
    [ 2,  2,  5,  5,  8,  8, 11, 11],
    [ 2,  2,  5,  5,  8,  8, 11, 11],
    [ 2,  2,  5,  5,  8,  8, 11, 11],
    [ 3,  3,  6,  6,  9,  9, 12, 12],
    [ 3,  3,  6,  6,  9,  9, 12, 12],
    [ 3,  3,  6,  6,  9,  9, 12, 12]])
rows%xrows = 0
cols%xcols = 0

rows is the number of rows of the array, cols is the number of columns of the array, xrows is the number of rows in each slice, xcols is the number of column in each slice. the answer should be general not for this parameters

could you be more specific on what each parameter is defining? — JackRaBeat
– JackRaBeat, Commented Apr 20, 2022 at 15:39

mozway · Accepted Answer · 2022-04-20 18:54:30Z

2

IIUC, you can use:

((np.arange(rows//xrows*cols//xcols, dtype=int)+1)
 .reshape((rows//xrows,cols//xcols), order='F')
 .repeat(xrows,0)
 .repeat(xcols,1)
 )

Output:

array([[ 1,  1,  4,  4,  7,  7, 10, 10],
       [ 1,  1,  4,  4,  7,  7, 10, 10],
       [ 1,  1,  4,  4,  7,  7, 10, 10],
       [ 2,  2,  5,  5,  8,  8, 11, 11],
       [ 2,  2,  5,  5,  8,  8, 11, 11],
       [ 2,  2,  5,  5,  8,  8, 11, 11],
       [ 3,  3,  6,  6,  9,  9, 12, 12],
       [ 3,  3,  6,  6,  9,  9, 12, 12],
       [ 3,  3,  6,  6,  9,  9, 12, 12]])

edited Apr 20, 2022 at 18:54

answered Apr 20, 2022 at 16:26

mozway

267k13 gold badges56 silver badges106 bronze badges

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

2 Comments

user18628648 Over a year ago

I can use that, but better if it possible to tune the slices by column not by rows

mozway Over a year ago

@user18628648 sure, that was an oversight, see update

Collectives™ on Stack Overflow

set values of 2D numpy array by parameters

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related