mirror of
https://github.com/heqin-zhu/algorithm.git
synced 2024-03-22 13:30:46 +08:00
157 lines
3.9 KiB
Python
157 lines
3.9 KiB
Python
''' mbinary
|
|
#########################################################################
|
|
# File : vector_norm.py
|
|
# Author: mbinary
|
|
# Mail: zhuheqin1@gmail.com
|
|
# Blog: https://mbinary.xyz
|
|
# Github: https://github.com/mbinary
|
|
# Created Time: 2018-10-02 21:14
|
|
# Description:
|
|
#########################################################################
|
|
'''
|
|
|
|
from random import randint, random
|
|
import numpy as np
|
|
from operator import neg, and_
|
|
from functools import reduce
|
|
|
|
|
|
class obj():
|
|
def __init__(self, data):
|
|
self.data = np.array(data)
|
|
|
|
def __add__(self, x):
|
|
data = x.data if self.__class__ == x.__class__ else x
|
|
return self.__class__(self.data + data)
|
|
|
|
def __radd__(self, x):
|
|
data = x.data if self.__class__ == x.__class__ else x
|
|
return self.__class__(data + self.data)
|
|
|
|
def __iadd__(self, x):
|
|
data = x.data if self.__class__ == x.__class__ else x
|
|
self.data += data
|
|
|
|
def __mul__(self, x):
|
|
data = x.data if self.__class__ == x.__class__ else x
|
|
return self.__class__(self.data * data)
|
|
|
|
def __imul__(self, x):
|
|
data = x.data if self.__class__ == x.__class__ else x
|
|
self.data *= data
|
|
|
|
def __rmul__(self, x):
|
|
data = x.data if self.__class__ == x.__class__ else x
|
|
return self.__class__(data * self.data)
|
|
|
|
def __neg__(self):
|
|
return neg(self)
|
|
|
|
def __abs__(self):
|
|
return abs(self.data)
|
|
'''
|
|
@property
|
|
def data(self):
|
|
return self.data
|
|
@data.setter
|
|
def data(self,s):
|
|
self.data = s
|
|
'''
|
|
|
|
def norm(self, n=0):
|
|
'''the default is +oo norm'''
|
|
absolute = abs(self.data)
|
|
if n < 1:
|
|
return max(absolute)
|
|
return (sum(absolute**n))**(1/n)
|
|
|
|
def hasNorm(self):
|
|
'''check norm's three necessary conditions:
|
|
1. not neg
|
|
2. homogenious (qici)
|
|
3. triangle inequlity
|
|
|
|
there is much probably wrong
|
|
'''
|
|
bl = reduce(and_, [self.norm(i) >= 0 for i in range(3)])
|
|
if bl:
|
|
n = randint(2, 100)
|
|
bl = reduce(and_, [n*(self.norm(i)) == (n*self).norm(i)
|
|
for i in range(3)])
|
|
if bl:
|
|
another = self*randint(2, 10)-randint(1, 100)
|
|
return reduce(and_, [(another+self).norm(i) <= another.norm(i)+self.norm(i) for i in range(3)])
|
|
return False
|
|
|
|
|
|
class vector(obj):
|
|
def __init__(self, arr):
|
|
''' arr: iterable'''
|
|
self.data = np.array(arr)
|
|
|
|
def innerProduct(self, x):
|
|
return sum(self.data*x)
|
|
|
|
def outerProduct(self, x):
|
|
pass
|
|
|
|
|
|
class matrix(obj):
|
|
def __init__(self, s):
|
|
'''s is a list of lists'''
|
|
self.data = np.mat(s)
|
|
self.T = None
|
|
self. I = None
|
|
'''
|
|
@property
|
|
def T(self):
|
|
if self.T==None:self.T = self.data.T
|
|
return self.T
|
|
@T.setter
|
|
def T(self,s):
|
|
self.T = s
|
|
@property
|
|
def I(self):
|
|
if self.I == None: self.I = self.data.I
|
|
return self.I
|
|
|
|
@I.setter
|
|
def I(self,s):
|
|
self.I = s
|
|
'''
|
|
|
|
def E(self, n=None):
|
|
if n is None:
|
|
n = self.data.shape[0]
|
|
return np.eye(n)
|
|
|
|
def norm(self, n=0):
|
|
absolute = abs(self.data)
|
|
if n < 1:
|
|
# max of one row sum
|
|
return max([sum(i) for i in absolute])
|
|
if n == 1:
|
|
return self.norm1()
|
|
elif n == 2:
|
|
return self.norm2()
|
|
|
|
def norm1(self):
|
|
''' max of sum of cols'''
|
|
absolute = abs(self.data)
|
|
return max(absolute.sum(axis=0))
|
|
|
|
def norm2(self):
|
|
''' max of sum of rows'''
|
|
absolute = abs(self.data)
|
|
return max(absolute.sum(axis=1))
|
|
|
|
def norm_f(self):
|
|
return sum((self.data**2).sum(axis=1))**0.5
|
|
|
|
|
|
if __name__ == '__main__':
|
|
v1 = vector([1, -2, 3, 4])
|
|
v2 = vector([0, 2, 0, 5])
|
|
m1 = matrix([v1, v2, v2, v1])
|
|
print([v1.norm(i) for i in range(3)])
|