Python - 数学 - 行列 - 行列とは何か？- 行列の表現間の変換(行列の列の辞書表現を返す1行プロシージャー(関数))

行列プログラマー
原著

開発環境

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

行列プログラマー(Philip N. Klein (著)、 松田 晃一 (翻訳)、 弓林 司 (翻訳)、 脇本 佑紀 (翻訳)、 中田 洋 (翻訳)、 齋藤 大吾 (翻訳)、オライリージャパン)の4章(行列)、4.1(行列とは何か？)、4.1.6(行列の表現間の変換)、クイズ4.1.10を取り組んでみる。

クイズ4.1.10

コード(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]}

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)
    A = mat2coldict(M)
    print(A)
    pprint.pprint(A)

入出力結果(Terminal, IPython)

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