numpy ndarray怎么用
时间: 2023-11-12 10:08:07 浏览: 177
NumPy是Python科学计算的核心库之一,它提供了一种高效的多维数组对象和基于数组的计算方法。ndarray是NumPy中最重要的类,它表示一个N维数组对象。
下面是一些ndarray的基本用法:
1. 创建一个数组
可以使用numpy.array()函数来创建一个数组,可以从Python列表、元组等对象中创建。
```python
import numpy as np
# 从Python列表中创建
a = np.array([1, 2, 3])
print(a) # [1 2 3]
# 从元组中创建
b = np.array((4, 5, 6))
print(b) # [4 5 6]
# 创建一个二维数组
c = np.array([[1, 2], [3, 4]])
print(c)
# [[1 2]
# [3 4]]
```
2. 数组的形状和维度
可以使用ndarray.shape属性来获取数组的形状。ndarray.ndim属性则可以获取数组的维度。
```python
import numpy as np
a = np.array([[1, 2], [3, 4], [5, 6]])
print(a.shape) # (3, 2)
print(a.ndim) # 2
```
3. 数组的索引和切片
可以使用索引和切片来访问数组中的元素。
```python
import numpy as np
a = np.array([1, 2, 3, 4, 5])
print(a[0]) # 1
print(a[1:3]) # [2 3]
b = np.array([[1, 2], [3, 4], [5, 6]])
print(b[1, 0]) # 3
print(b[:, 1]) # [2 4 6]
```
4. 数组的运算
可以使用NumPy提供的函数进行数组的运算。
```python
import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
print(a + b) # [5 7 9]
print(a * b) # [ 4 10 18]
c = np.array([[1, 2], [3, 4]])
d = np.array([[5, 6], [7, 8]])
print(c + d)
# [[ 6 8]
# [10 12]]
print(c * d)
# [[ 5 12]
# [21 32]]
```
这只是ndarray的一些基本用法,NumPy还提供了很多其他的功能,例如数组的广播、矩阵运算、随机数生成等。如果你想深入了解NumPy的使用方法,可以参考官方文档:https://numpy.org/doc/stable/。
阅读全文