تبدیل ماتریس در پایتون

Question

سلام وقتتون بخیر ، دوستان من یه ماتریس دارم  با ابعاد 8064*40*40 میخوام درایه های (:,1,1)  از این ماتریس استخراج کنم، کسی میتونه راهنماییم کنه که چطوری میتونم این کار رو انجام بدم؟

Answer 1

import  numpy as np

a = np.zeros(shape=(8,4,4))
for i in range(a.shape[0]):
   a[i,0,0] = 1

print(a)