NumPy 是 Python 中一个功能强大的数值计算库，它提供了强大的多维数组对象和一系列数学函数。以下是一些 NumPy 的基本概念和用法。

安装 NumPy

首先，您需要安装 NumPy。可以通过以下命令安装：

pip install numpy

创建数组

NumPy 提供了多种创建数组的方法。以下是一些常见的创建数组的方法：

  • numpy.array(): 创建一个数组。
  • numpy.zeros(): 创建一个元素全部为零的数组。
  • numpy.ones(): 创建一个元素全部为一的数组。
import numpy as np


a = np.array([1, 2, 3, 4])

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

# 创建一个三维数组
c = np.ones((2, 2, 2))

数组索引

NumPy 支持数组索引，您可以使用与 Python 列表相似的方式来访问数组中的元素。

# 访问数组中的元素
print(a[0])  # 输出：1

# 修改数组中的元素
a[0] = 10
print(a)  # 输出：[10  2  3  4]

数组操作

NumPy 提供了丰富的数组操作功能，以下是一些常见的操作：

  • 索引和切片
  • 数组运算
  • 矩阵运算
# 索引和切片
print(a[1:3])  # 输出：[2 3]

# 数组运算
print(a + 1)  # 输出：[11  3  4  5]

# 矩阵运算
print(np.dot(a, b))  # 输出：[ 8 10]

NumPy 数组与 Python 列表的区别

NumPy 数组与 Python 列表有一些区别：

  • NumPy 数组是固定类型的，而 Python 列表可以是不同类型的。
  • NumPy 数组操作通常比 Python 列表操作要快得多。

更多关于 NumPy 的内容，您可以访问本站的 NumPy 教程。

[center]NumPy Logo