MathLib/statistics/python/src/Vector.py at master · jingyuexing/MathLib · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
# -*- coding: utf-8 -*-
# @Author: Jingyuexing
# @Date:   2019-07-12 00:24:07
# @Last Modified by:   admin
# @Last Modified time: 2021-06-24 20:58:06


class Vector:
    """docstring for Vector
    向量
    """

    def __init__(self, data=[]):
        if isinstance(data, list):
            self.data = data
        else:
            print("Error:data type is not list")

    def add(self, data):
        '''向量相加

        [description]

        Arguments:
            data {Vector} -- 相加的向量

        Returns:
            list -- 向量和值
        '''
        tempAry = []
        if isinstance(data, Vector) and len(data.data) == len(self.data):
            for i in range(0, len(data.data)):
                tempAry.append(self.data[i] + data.data[i])
        return tempAry

    def mult(self, data):
        """[对向量两个向量进行求差]

        [description]

        Arguments:
            data {[Vector]} -- [相减向量]

        Returns:
            [list] -- [向量和结果值]
        """
        tempAry = []
        if isinstance(data, Vector) and len(data.data) == len(self.data):
            for i in range(0, len(data.data)):
                tempAry.append(data.data[i] - self.data[i])
            return tempAry
        else:
            print("Error:参数必须是向量实例或两个向量长度需相同")

    def mod(self, data):
        '''向量求模

        [description]

        Arguments:
            data {Vecotr} -- 向量

        Returns:
            number -- 模值
        '''
        total = 0
        if isinstance(data, Vector):
            for i in range(0, len(data.data)):
                total = total + data.data[i]**2
            return total**(1 / 2)
        else:
            return TypeError(data + "is not Vector")

    def product(self, data):
        '''向量乘积

        >>>Vector_1 = Vector([2,3,4,5,6,7])
        >>>Vector_2 = Vector([12,5,78,2,12])
        >>>Vector_2.product(Vector_1) return 748:number
        >>>Vector_2.product(8)  return [96, 40, 624, 16, 96, 360]
        Arguments:
            data {number | list | Vector} -- 求点积或者向量乘积

        Returns:
            number|list -- 向量乘积值
        '''
        total = 0
        if isinstance(data, int) or isinstance(data, float):
            for i in range(0, len(self.data)):
                self.data[i] = self.data[i] * data
        if isinstance(data, list):
            if len(self.data) == data.__len__():
                for i in range(0, len(data)):
                    total = total + self.data[i] * data[i]
                return total
            else:
                return TypeError("参数长度不一致!")
        if isinstance(data, Vector):
            return self.product(data.data)

    def isVertical(self, data):
        '''判断是否和另外的向量垂直,垂直返回真,不垂直返回假


        Arguments:
            data { Vector|List<number> } --  和自身比较的向量

        Returns:
            bool --   是否垂直
        '''
        if isinstance(data, list) and self.product(data) == 0:
            return True
        if isinstance(data, Vector):
            self.isVertical(data.data)
        return False

    def map(self):
        return self.data

    def angle(self, data):
        '''向量夹角
        Arguments:
            data {  Vector | list<number>} --   向量或者向量数据

        Returns:
            number -- 结果值
        '''
        if isinstance(data, list):
            total = 0
            if len(data) == len(self.data):
                for i in range(0, len(data)):
                    total = total + self.data[i] * data[i]
            return total / self.mod(data) * self.mod(self.data)
        if isinstance(data, Vector):
            self.angle(data.data)

    def insertData(self, *args):
        for i in args:
            self.data.append(i)
        return self

    def fill(self, value):
        """[summary]

        [description]

        Arguments:
            value {[type]} -- [description]

        Returns:
            [type] -- [description]
        """
        for x in range(len(self.data)):
            self.data[x] = value
        return self