from mrjob.job import MRJob
class MrMatrixMultiplier(MRJob):
def mapper_init(self):
self.ROWS = 0
self.COLS = 0
self.kvpairs = []
def mapper(self, _, line):
line_list = line.split(' ')
if (len(line_list) == 2):
[self.ROWS, self.COLS] = line_list
else:
self.i = line_list[0]
self.j = line_list[1]
self.v = line_list[2]
for k in range(1, int(self.ROWS)):
print [[self.i, k],['A', self.j, self.v]]
def reducer(self, key, values):
yield key, sum(values)
if __name__ == '__main__':
MrMatrixMultiplier.run()