Python - 数学 - 行列 - ベクトルとしての行列

行列プログラマー
原著

開発環境

• macOS Sierra - Apple (OS)
• Emacs (Text Editor)
• Python 3.6 (プログラミング言語)

行列プログラマー(Philip N. Klein (著)、 松田 晃一 (翻訳)、 弓林 司 (翻訳)、 脇本 佑紀 (翻訳)、 中田 洋 (翻訳)、 齋藤 大吾 (翻訳)、オライリージャパン)の4章(行列)、4.3(ベクトルとしての行列)、クイズ4.3.1を取り組んでみる。

クイズ4.3.1

コード(Emacs)

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from vec import Vec
import pprint


class Mat:

    def __init__(self, labels, function):
        self.D = labels
        self.f = function

    def __str__(self):
        return 'Mat({0}, {1})'.format(labels, function)


def mat2rowdict(A):
    return {i: Vec(A.D[1], {j: A.f[(i, j)] for j in A.D[1]}) for i in A.D[0]}


def mat2coldict(A):
    return {j: Vec(A.D[0], {i: A.f[(i, j)] for i in A.D[0]}) for j in A.D[1]}


def mat2vec(M):
    return Vec({(i, j) for i in M.D[0] for j in M.D[1]}, M.f)

if __name__ == '__main__':
    labels = ({'a', 'b'}, {'@', '#', '?'})
    function = {('a', '@'): 1, ('a', '#'): 2, ('a', '?'): 3,
                ('b', '@'): 10, ('b', '#'): 20, ('b', '?'): 30}

    M = Mat(labels, function)
    print(M)
    v = mat2vec(M)
    print(v)

入出力結果(Terminal, IPython)

$ ./sample8.py
Mat(({'a', 'b'}, {'@', '#', '?'}), {('a', '@'): 1, ('a', '#'): 2, ('a', '?'): 3, ('b', '@'): 10, ('b', '#'): 20, ('b', '?'): 30})

 ('a', '#') ('a', '?') ('a', '@') ('b', '#') ('b', '?') ('b', '@')
------------------------------------------------------------------
          2          3          1         20         30         10
$