NumPy 是 Python 中一个强大的数学库,主要用于数值计算。以下是一些 NumPy 的基本教程,帮助你快速上手。

安装 NumPy

首先,你需要安装 NumPy。可以通过以下命令进行安装:

pip install numpy

基本操作

创建数组

NumPy 提供了多种创建数组的方法,以下是一些常用的方法:

  • numpy.array(): 创建一个数组。
  • numpy.zeros(): 创建一个全零数组。
  • numpy.ones(): 创建一个全一数组。
  • numpy.empty(): 创建一个未初始化的数组。
import numpy as np

# 创建一个一维数组
a = np.array([1, 2, 3, 4, 5])

# 创建一个二维数组
b = np.zeros((3, 4))

# 创建一个未初始化的数组
c = np.empty((2, 3))

数组索引

NumPy 数组支持类似 Python 列表的索引方式,以下是一些常用的索引方法:

  • a[0]: 获取第一个元素。
  • a[1:3]: 获取第二个和第三个元素。
  • a[:]: 获取整个数组。
print(a[0])  # 输出:1
print(a[1:3])  # 输出:[2 3]
print(a[:])  # 输出:[1 2 3 4 5]

数组操作

NumPy 提供了丰富的数组操作方法,以下是一些常用的操作:

  • a + b: 数组加法。
  • a * b: 数组乘法。
  • a ** b: 数组幂运算。
  • a < b: 数组比较。
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]

# 数组幂运算
print(a ** b)  # 输出:[1 32 729]

# 数组比较
print(a < b)  # 输出:[False False True]

学习资源

更多关于 NumPy 的教程和资源,请访问以下链接:

NumPy Logo