np.array()用法
时间: 2023-09-26 11:09:20 浏览: 53
`np.array()` 是 NumPy 库中的一个函数,用于创建数组。它接受一个列表、元组、数组或其他序列类型作为参数,并返回一个 NumPy 数组对象。
以下是一些常用的用法:
1. 从列表或元组创建一维数组:
```python
import numpy as np
list1 = [1, 2, 3]
arr1 = np.array(list1)
print(arr1) # [1 2 3]
tuple1 = (4, 5, 6)
arr2 = np.array(tuple1)
print(arr2) # [4 5 6]
```
2. 从列表嵌套创建多维数组:
```python
import numpy as np
list2 = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
arr3 = np.array(list2)
print(arr3)
"""
[[1 2 3]
[4 5 6]
[7 8 9]]
"""
```
3. 创建全零/全一数组:
```python
import numpy as np
zeros_arr = np.zeros((2, 3)) # 2行3列的全零数组
print(zeros_arr)
"""
[[0. 0. 0.]
[0. 0. 0.]]
"""
ones_arr = np.ones((3, 2)) # 3行2列的全一数组
print(ones_arr)
"""
[[1. 1.]
[1. 1.]
[1. 1.]]
"""
```
4. 创建指定范围内的数组:
```python
import numpy as np
range_arr = np.arange(0, 10, 2) # 从0开始,步长为2,直到小于10
print(range_arr) # [0 2 4 6 8]
linspace_arr = np.linspace(0, 1, 5) # 在0到1之间生成5个等距数字
print(linspace_arr) # [0. 0.25 0.5 0.75 1. ]
```
这些只是 `np.array()` 的一些常见用法,它还有很多其他参数和选项。如果想要了解更多,请查看 NumPy 官方文档。
阅读全文