两个一维numpy数组相乘得到两维数组
时间: 2023-10-01 18:07:36 浏览: 142
利用numpy实现一、二维数组的拼接简单代码示例
如果你想将两个一维NumPy数组相乘得到一个二维数组,可以使用NumPy的outer函数。outer函数将两个一维数组作为输入,然后输出一个二维数组,其中每个元素是两个输入数组的元素乘积。具体来说,输出的数组的形状是第一个输入数组的长度乘以第二个输入数组的长度。
以下是一个示例代码,演示如何使用outer函数将两个一维数组相乘得到一个二维数组:
```python
import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
result = np.outer(arr1, arr2)
print(result)
```
输出结果为:
```
[[ 4 5 6]
[ 8 10 12]
[12 15 18]]
```
在这个示例中,我们首先创建了两个一维数组`arr1`和`arr2`。然后,我们使用NumPy的outer函数将这两个数组相乘,得到一个二维数组`result`。`result`中的每个元素都是`arr1`和`arr2`中对应位置上的元素的乘积。需要注意的是,`result`的形状是3x3,因为`arr1`和`arr2`都有3个元素。
阅读全文