如何用python创建一个维度(3,1)的数组,元素为任意复数的数组
时间: 2023-01-31 19:09:42 浏览: 63
可以使用NumPy库中的函数`numpy.array()`来创建一个维度为(3,1)的数组。例如,要创建一个元素为任意复数的数组,可以这样做:
```
import numpy as np
arr = np.array([[1+2j], [3+4j], [5+6j]])
print(arr)
```
输出结果为:
```
[[1.+2.j]
[3.+4.j]
[5.+6.j]]
```
也可以使用NumPy的矩阵创建函数`numpy.matrix()`来创建一个矩阵,例如:
```
import numpy as np
mat = np.matrix([[1+2j], [3+4j], [5+6j]])
print(mat)
```
输出结果为:
```
[[1.+2.j]
[3.+4.j]
[5.+6.j]]
```
注意,NumPy中的矩阵是二维的,而数组是可以有任意维度的。因此,如果想要创建具有更多维度的数组,可以使用`numpy.array()`函数。
例如,要创建一个3x3x3的复数数组,可以这样做:
```
import numpy as np
arr = np.array([[[1+2j, 3+4j, 5+6j],
[7+8j, 9+10j, 11+12j],
[13+14j, 15+16j, 17+18j]],
[[19+20j, 21+22j, 23+24j],
[25+26j, 27+28j, 29+30j],
[31+32j, 33+34j, 35+36j]],
[[37+38j, 39+40j, 41+42j],
[43+44j, 45+46j, 47+48j],
[49+50j, 51+52j, 53+54j]]])
print(arr)
```
输出结果为:
```
[[[ 1.+2.j 3.+