2012-03-01 4 views
3

하나의 "화려한"조각을 사용하여 행렬에서 행과 열을 추출하려면이 방법이 있습니까?간단한 행과 열 추출, numpy

m = matrix([[1, 2, 3], 
      [4, 5, 6], 
      [7, 8, 9]]) 

내 목표는 내가

d = [0,2] 

내가

m[d][:,d] 

하여 기능을 달성 할 수 원하는 내가 항목의 목록을

matrix([[1, 3], 
     [7, 9]]) 

이다 그러나이 더 간단한 표현?

답변

8

당신은 numpy.ix_를 사용하여이 작업을 수행 할 수 있습니다

m = matrix([[1, 2, 3], 
      [4, 5, 6], 
      [7, 8, 9]]) 

d = [0,2] 
print m[ix_(d,d)] 

이 방출되는 :

[[1 3] 
[7 9]]