큰 2D 배열의 각 2D 사각형 서브 어레이의 대각선 추출

Question

이것은 내 첫 번째 프로그래밍 클래스이며 데이터 과학을위한 Python을 배우게되어 매우 기쁩니다. 행렬의 모든 대각선 번호를 반환하는 루프를 작성하는 방법을 알아낼 수 없습니다. 아래 코드는 내가 얼마나 가깝거나 멀리 떨어져 있는가? 고맙습니다!

import numpy as np 
cols = 0 

matrixA = np.array([[2,0,0], [0,3,0], [0,0,4], [6,0,0], [0,7,0], [0,0,8]]) 

for rows in range(6): 
    if rows == cols: 
     print(matrixA[rows, cols]) 
    cols = cols + 1 

Answer 1

현재 솔루션은 작동하지 않습니다. 당신은 당신의 색인이 범위를 벗어나지 않도록주의해야합니다. 를 실행하면 제공 :

IndexError: index 3 is out of bounds for axis 1 with size 3 

cols 여기에 걸릴 허용되는 최대 값은 2 때문입니다.

대안으로

---

, 당신은 np.diag 사용할 수 있습니다 당신은 1D 결과를 원하는 경우

print(x) 
array([[2, 0, 0], 
     [0, 3, 0], 
     [0, 0, 4], 
     [6, 0, 0], 
     [0, 7, 0], 
     [0, 0, 8]]) 

res = np.array([np.diag(x, -offset) for offset in range(0, *x.shape)]) 

print(res) 
array([[2, 3, 4], 
     [6, 7, 8]]) 

이 np.ravel 전화 :

print(res.ravel()) 
array([2, 3, 4, 6, 7, 8]) 

Answer 2

간단한 작업을 수행하려면 numpy과 같은 무거운 라이브러리가 필요하지 않습니다. 일반 파이썬에서는이 같은 zip 및 itertools.cycle(...)를 사용을 통해 작업을 수행 할 수 있습니다

>>> from itertools import cycle 

>>> my_list = [[2,0,0], [0,3,0], [0,0,4], [6,0,0], [0,7,0], [0,0,8]] 
>>> for i, j in zip(my_list, cycle(range(3))): 
...  print(i[j]) 
... 
2 
3 
4 
6 
7 
8 

Answer 3

왜 COLS 전혀 있나요? 항상 행과 같습니다. 맞습니까? 이 계정에 matrixA가 사각형이 있다는 사실을 고려하지 않기 때문에

for rows in range(6): 
    print(matrixA[rows,rows])