Python - 数学 - ベクトル - ベクトルの加法(平行移動とベクトルの加法)

行列プログラマー
原著

開発環境

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

行列プログラマー(Philip N. Klein (著)、 松田 晃一 (翻訳)、 弓林 司 (翻訳)、 脇本 佑紀 (翻訳)、 中田 洋 (翻訳)、 齋藤 大吾 (翻訳)、オライリージャパン)の1章(ベクトル)、2.4(ベクトルの加法)、2.4.1(平行移動とベクトルの加法)、クイズ 2.4.2、課題 2.4.3、クイズ 2.4.4を取り組んでみる。

クイズ 2.4.2、課題 2.4.3、クイズ 2.4.4

コード(Emacs)

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

import unittest


def add2(v, w):
    return [a + b for (a, b) in zip(v, w)]

# 2.4.2


def f(v):
    return add2(v, [1, 2])


# 2.4.3

# 2.4.4
def addn(v, w):
    return [a + w[i] for i, a in enumerate(v)]


class Test(unittest.TestCase):

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def test_2_1(self):
        self.assertEqual(f([4, 4]), [5, 6])

    def test_2_2(self):
        self.assertEqual(f([-4, -4]), [-3, -2])

    def test_4_1(self):
        self.assertEqual(addn([4, 4], [1, 2]), [5, 6])

    def test_4_2(self):
        self.assertEqual(addn([-4, -4], [1, 2]), [-3, -2])

    def test4_3(self):
        self.assertEqual(addn([], []), [])

    def test4_3(self):
        self.assertEqual(addn([1, 2, 3, 4, 5], [6, 7, 8, 9, 0]),
                         [7, 9, 11, 13, 5])


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

入出力結果(Terminal, IPython)

$ ./sample4_2.py -v
test4_3 (__main__.Test) ... ok
test_2_1 (__main__.Test) ... ok
test_2_2 (__main__.Test) ... ok
test_4_1 (__main__.Test) ... ok
test_4_2 (__main__.Test) ... ok

----------------------------------------------------------------------
Ran 5 tests in 0.000s

OK
$