niklas/pwr_project
1
0
Fork 0
Code Issues Activity

Minor code cleanup

Browse Source
This commit is contained in:
Niklas Birk
2024-02-22 02:28:26 +01:00
parent 3ec62d99f7
commit 3a4c0ed75d
3 changed files with 19 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/matrix.py
View File

@ -1,7 +1,6 @@
import math
import numpy
from numpy import linalg
class Matrix:
@ -76,9 +75,8 @@ class Matrix:
self.__shape__ = (len(data), len(data[0]))
self.__data__ = data
else:
raise ValueError(
"Only following signatures are allowed: "
"(list), (numpy.ndarray), (list, tuple), (list, str, int), (list, str, int, int), (str, int)")
raise ValueError("Only following signatures are allowed: "
"(list), (numpy.ndarray), (list, tuple), (list, str, int), (list, str, int, int), (str, int)")
def get_data(self):
"""
@ -243,7 +241,7 @@ class Matrix:
abs_sum = 0
for i in range(rows):
for j in range(cols):
abs_sum += abs(self.__data__[i][j])**2
abs_sum += abs(self.__data__[i][j]) ** 2
norm = math.sqrt(abs_sum)
elif f == "col sum":
row_sum = [0 for _ in range(cols)]