NumPy 是 Python 中一个强大的库，主要用于处理大型多维数组以及进行数学计算。本教程将带你入门 NumPy 的基本使用。

安装 NumPy

首先，确保你已经安装了 NumPy。你可以使用以下命令进行安装：

pip install numpy

基本数据类型

NumPy 支持多种数据类型，以下是一些常用的类型：

  • int：整数
  • float：浮点数
  • bool：布尔值

创建数组

你可以使用 numpy.array() 函数来创建一个数组：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(arr)

输出：

[1 2 3 4 5]

数组操作

NumPy 提供了丰富的数组操作函数，以下是一些示例：

  • numpy.sum()：计算数组所有元素的和
  • numpy.mean()：计算数组所有元素的平均值
  • numpy.max()：返回数组中的最大值
print(np.sum(arr))
print(np.mean(arr))
print(np.max(arr))

输出：

15
3.0
5

数组索引

你可以使用索引来访问数组中的元素：

print(arr[0])  # 输出：1
print(arr[1:3])  # 输出：[2 3]

高级操作

NumPy 还支持更高级的操作，例如：

  • numpy.dot()：计算两个数组的点积
  • numpy.linalg.inv()：计算矩阵的逆
import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])

print(np.dot(arr1, arr2))
print(np.linalg.inv(arr1))

输出：

[[19 22]
 [43 50]]
[[ 0.4 -0.2]
 [-0.2   0.4]]

学习资源

想要了解更多关于 NumPy 的知识，可以访问我们的 NumPy 教程。

图片

NumPy 图标

NumPy_Icon