numpy中python dot使用技巧

python

小樊

103

2024-06-15 12:21:29

栏目: 编程语言

在numpy中，我们可以使用numpy.dot()函数来计算两个数组的点积。点积是两个数组的对应元素相乘后再相加的结果。下面是几个使用numpy.dot()函数的技巧：

计算两个一维数组的点积：

import numpy as np

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

result = np.dot(a, b)
print(result)  # Output: 32

计算两个二维数组的点积：

import numpy as np

a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])

result = np.dot(a, b)
print(result)
# Output:
# [[19 22]
#  [43 50]]

计算一个二维数组和一个一维数组的点积：

import numpy as np

a = np.array([[1, 2], [3, 4]])
b = np.array([5, 6])

result = np.dot(a, b)
print(result)  # Output: [17 39]

注意，点积的计算规则是第一个数组的最后一个维度与第二个数组的倒数第二个维度相等。因此，要确保数组的维度匹配才能正确计算点积。

numpy中python dot使用技巧

最新问答

相关标签