Как умножить 2 матрицы в numpy: простое руководство + 🧮 = 🚀
import numpy as np
# Создание двух матриц
matrix1 = np.array([[1, 2], [3, 4]])
matrix2 = np.array([[5, 6], [7, 8]])
# Умножение матриц
result = np.dot(matrix1, matrix2)
print(result)
В данном примере создаются две матрицы "matrix1" и "matrix2". Затем с помощью функции np.dot() умножаются эти матрицы, и результат сохраняется в переменной "result". Наконец, результат выводится на экран при помощи функции print(). Убедитесь, что размерности матриц совместимы для умножения.
Детальный ответ
Как умножить 2 матрицы в библиотеке NumPy
Прежде чем мы начнем, давайте вспомним основы умножения матриц. Для умножения двух матриц их размерности должны быть согласованы. Если у нас есть матрица A размером MxN и матрица B размером NxP, то произведение этих матриц будет матрицей C размером MxP, где каждый элемент C[i][j] вычисляется следующим образом:
C[i][j] = sum(A[i][k] * B[k][j] for k in range(N))
Теперь, когда мы понимаем основы, мы можем перейти к использованию библиотеки NumPy для умножения двух матриц.
Во-первых, нам необходимо импортировать библиотеку NumPy:
import numpy as np
Затем мы можем создать наши две матрицы с помощью функции numpy.array()
. Предположим, что у нас есть матрица A размером 2x3:
A = np.array([[1, 2, 3],
[4, 5, 6]])
и матрица B размером 3x2:
B = np.array([[7, 8],
[9, 10],
[11, 12]])
Теперь, чтобы умножить эти две матрицы, мы можем использовать функцию numpy.dot()
:
C = np.dot(A, B)
Матрица C будет содержать результат умножения матрицы A на матрицу B.
Давайте выведем результат:
print(C)
Ожидаемый вывод:
[[ 58 64]
[139 154]]
В этом примере матрица A имеет размер 2x3, а матрица B имеет размер 3x2. Результирующая матрица C имеет размер 2x2. Каждый элемент C[i][j] вычисляется с помощью суммирования произведений элементов соответствующих строк матрицы A и столбцов матрицы B.
Если вам нужно умножить больше чем две матрицы, вы можете использовать функцию numpy.matmul()
. Она позволяет умножать несколько матриц за одну операцию.
D = np.matmul(A, np.matmul(B, C))
В этом примере мы сначала умножаем матрицы B и C, а затем умножаем результат на матрицу A.
Вот и все! Теперь вы знаете, как умножать две матрицы с помощью библиотеки NumPy. Удачи в ваших вычислениях!