Python - 数学 - ベクトル - ベクトルの和とスカラーとの積の組み合わせ - はじめての凸結合

行列プログラマー
原著

開発環境

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

行列プログラマー(Philip N. Klein (著)、 松田 晃一 (翻訳)、 弓林 司 (翻訳)、 脇本 佑紀 (翻訳)、 中田 洋 (翻訳)、 齋藤 大吾 (翻訳)、オライリージャパン)の1章(ベクトル)、2.6(ベクトルの和とスカラーとの積の組み合わせ)、2.6.3(はじめての凸結合)、課題2.6.9を取り組んでみる。

課題2.6.9

コード(Emacs)

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

from plotting import plot


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


def muln(v, w):
    return [a * w[i] for i, a in enumerate(v)]


def scalar_vector_mult(alpha, v):
    return [alpha * x for x in v]


def segment(pt1, pt2):
    n = 100
    return [
        addn(scalar_vector_mult(alpha / n, pt1),
             scalar_vector_mult((n - alpha) / n, pt2))
        for alpha in range(n)
    ]

import unittest


class Test(unittest.TestCase):

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def test_count(self):
        seg = segment([1, 2, 3], [4, 5, 6])
        self.assertEqual(len(seg), 100)

if __name__ == '__main__':
    plot(segment([3.5, 3], [0.5, 1]), 10)
    input()
    unittest.main()

入出力結果(Terminal, IPython)

$ ./sample6_9.py -v
q
test_count (__main__.Test) ... ok

----------------------------------------------------------------------
Ran 1 test in 0.000s

OK
$