Python - 数学 - ベクトル - 辞書によるベクトルの表現(スカラーとベクトルの積)

行列プログラマー
原著

開発環境

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

行列プログラマー(Philip N. Klein (著)、 松田 晃一 (翻訳)、 弓林 司 (翻訳)、 脇本 佑紀 (翻訳)、 中田 洋 (翻訳)、 齋藤 大吾 (翻訳)、オライリージャパン)の1章(ベクトル)、2.7(辞書によるベクトルの表現)、2.7.2(スカラーとベクトルの積)、クイズ 2.7.3を取り組んでみる。

クイズ 2.7.3

コード(Emacs)

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

import unittest


class Vector:

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


def zero_vector(D):
    labels = D
    function = {k: 0 for k in labels}
    return Vector(labels, function)


def get_item(v, d):
    return v.f.get(d, 0)


def scalar_mul(v, alpha):
    return Vector(v.D, {d: x * alpha for d, x in v.f.items()})


class Test(unittest.TestCase):

    def setUp(self):
        self.function = {'A': 1, 'B': 2, 'C': 10}
        self.D = {'A', 'B', 'C', 'D'}
        self.v = Vector({'A', 'B', 'C', 'D'}, {'A': 1, 'B': 2, 'C': 10})

    def tearDown(self):
        pass

    def test_0(self):
        v = scalar_mul(self.v, 0)
        self.assertEqual(v.f, {'A': 0, 'B': 0, 'C': 0})
        self.assertEqual(self.v.f, self.function)

    def test_1(self):
        v = scalar_mul(self.v, 1)
        self.assertEqual(v.f, {'A': 1, 'B': 2, 'C': 10})
        self.assertEqual(self.v.f, self.function)

    def test_10(self):
        v = scalar_mul(self.v, 10)
        self.assertEqual(v.f, {'A': 10, 'B': 20, 'C': 100})
        self.assertEqual(self.v.f, self.function)


if __name__ == '__main__':
    unittest.main()

入出力結果(Terminal, IPython)

$ ./sample7_3.py -v
test_0 (__main__.Test) ... ok
test_1 (__main__.Test) ... ok
test_10 (__main__.Test) ... ok

----------------------------------------------------------------------
Ran 3 tests in 0.000s

OK
$