Pomoc z uproszczeniem kodu / optymalizacja kodu

Question

Cześć,
zastanawiam się czy mogę jakoś jeszcze uprościć kod programu/funkcji do mnożenia macierzy przez liczbę

def mnozenie_macierzy(macierz,liczba):
    for m in macierz:
        for n in m:
            m[n] *= liczba

    print(macierz)

tablica = [[0, 1, 2],
           [0, 1, 2],
           [0, 1, 2]]

a = 5

mnozenie_macierzy(tablica,a)

 

Answer 1

Najkrótsza wersja z użyciem numpy

import numpy as np

def mnozenie_macierzy(macierz,liczba):
    print(macierz * liczba)


tablica = np.array(
            [[0, 1, 2],
            [0, 1, 2],
            [0, 1, 2]])
a = 5

mnozenie_macierzy(tablica,a)

 

Answer 2

Jeśli możesz użyć: numpy, to masz zapis  [ on-line ]

import numpy as np

def mnozenie_macierzy(macierz, liczba):
    m = np.matrix(macierz)
    print(m * liczba)

 
tablica = [[0, 1, 2],
           [0, 1, 2],
           [0, 1, 2]]
a = 5
 
mnozenie_macierzy(tablica, a)

lub [ on-line ]

import numpy as np

def mnozenie_macierzy(macierz, liczba):
    print(np.matrix(macierz) * liczba)

 
tablica = [[0, 1, 2],
           [0, 1, 2],
           [0, 1, 2]]
a = 5
 
mnozenie_macierzy(tablica, a)

 

Answer 3

Jeśli bez numpy, to można choćby tak:

#!/usr/bin/env python3

def matrix_mul(matrix, value):
    return [[val * value for val in row] for row in matrix]
 
table = [[0, 1, 2],
           [0, 1, 2],
           [0, 1, 2]]
 
value = 5
 
print(matrix_mul(table, value))