np.dot()用法
时间: 2023-10-30 13:58:44 浏览: 129
np.dot()函数的用法详解
np.dot()是NumPy库中的一个函数,用于计算两个数组的点积(内积)。它可以用于计算矩阵乘法,向量点积以及其他多维数组的乘法操作。
使用方式如下:
```python
import numpy as np
# 创建两个数组
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
# 计算点积
result = np.dot(a, b)
print(result)
```
输出结果为:32
在此例子中,np.dot()函数将数组a和数组b进行点积运算,结果为32。
除了一维数组外,np.dot()还适用于多维数组的点积计算。例如,对于矩阵的乘法操作:
```python
matrix_a = np.array([[1, 2], [3, 4]])
matrix_b = np.array([[5, 6], [7, 8]])
result_matrix = np.dot(matrix_a, matrix_b)
print(result_matrix)
```
输出结果为:
```
[[19 22]
[43 50]]
```
在这个例子中,np.dot()函数用于计算矩阵matrix_a和matrix_b的乘积。输出结果是一个新的矩阵result_matrix。
总结来说,np.dot()函数可以用于计算两个数组的点积、向量的点积以及多维数组的乘法操作。
阅读全文