2016-05-25 49 views
0

我已经问过这个question但仍然无法弄清楚如何实现这个。如何实现m [1:3,0:2] + = 1等子矩阵的接口?

我有矩阵类:

class Matrix(list): 
    def __getitem__(self, item): 
     try: 
      return list.__getitem__(self, item) 
     except TypeError: 
      rows, cols = item 
      return [row[cols] for row in self[rows]] 

它允许做这样的事情:

m = Matrix([[i+j for j in [0,1,2,3]] for i in [0,4,8,12]]) 
print(m[0:2, 0:2]) 
will print: [[0, 1], [4, 5]] 

我也希望能够通过给定值增加/乘法子矩阵元素,如:

m[0:2, 0:2] += 1 
print(m[0:2, 0:2]) 
should print: [[1, 2], [5, 6]] 

我想实现这些方法:__add____setitem__

def __setitem__(self, key, value): 
    print(key, value) 

def __add__(self, item): 
    print(item) 
    for i in range(self): 
     for j in range(self[0]): 
      self[i][j] += item 

至少我想看到他们打印的东西。但它没有发生。我想用这样的例子:

m[1:2, 2:3] = m[1:2, 2:3] + 1 

,并得到一个错误,如:TypeError: can only concatenate list (not "int") to list

所以,我甚至没有魔法。呼叫失败之前。这个怎么做?

+0

你为什么要重塑numpy的? – shx2

+0

@ shx2学习目的 –

回答

1

你必须从 “的GetItem” 返回Matrix对象

class Matrix(list): 
    def __getitem__(self, item): 
     print "get" 
     try: 
      return Matrix(list.__getitem__(self, item)) 
     except TypeError: 
      rows, cols = item 
      return Matrix([row[cols] for row in self[rows]]) 
    def __setitem__(self, key, value): 
     print(key, value) 

    def __add__(self, item): 
     print "messi the great" 
     print(item) 
     # for i in range(self): 
     #  for j in range(self[0]): 
     #   self[i][j] += item 
m = Matrix([[i+j for j in [0,1,2,3]] for i in [0,4,8,12]]) 
print m[1:2, 2:3] 
m[1:2, 2:3] = m[1:2, 2:3] + 1 
相关问题