how to get subarray of matrix in python

Question

say I have the following array j:

[[1, 2, 3, 4, 5], 
 [7, 7, 7, 6, 4], 
 [1, 1, 2, 0, 0]]

how can I get the subarray of 2x2 so the subarray would be:

[[1, 2], 
 [7, 7],]

intuitively I assumed j[0:2][0:2] would do the trick but I get:

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

@mkrieger1 my fault I read the question wrong - removed the tag — Chris
– Chris, Commented May 26, 2022 at 19:43
Your use of array, subarray and matrix suggest you are using numpy. But the examples are indistinguishable from a list of lists. — hpaulj
– hpaulj, Commented May 26, 2022 at 19:44

Scott Hunter · Accepted Answer · 2022-05-26 19:38:41Z

2

You need to explicitly say what you want from each row:

[r[0:2] for r in j[0:2]]

answered May 26, 2022 at 19:38

Scott Hunter

50k12 gold badges65 silver badges107 bronze badges

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

Comments

eshirvana · Accepted Answer · 2022-05-26 19:42:48Z

2

in numpy you can do this:

import numpy as np 
j = np.array([[1, 2, 3, 4, 5], 
 [7, 7, 7, 6, 4], 
 [1, 1, 2, 0, 0]])

j[:2, :2]

output:

>>
[[1 2]
 [7 7]]

answered May 26, 2022 at 19:42

eshirvana

24.7k3 gold badges28 silver badges43 bronze badges

AboAmmar · Accepted Answer · 2022-05-26 19:58:31Z

0

You could also do:

[j[i][:2] for i in range(2)]

[[1, 2], 
 [7, 7]]

answered May 26, 2022 at 19:58

AboAmmar

5,5892 gold badges15 silver badges25 bronze badges