NumPy 是 Python 中一个用于数值计算的库,它提供了大量的数值计算函数和矩阵操作工具。下面是一些基础的 NumPy 教程内容。

安装 NumPy

在开始之前,确保你已经安装了 NumPy。你可以通过以下命令来安装:

pip install numpy

基础操作

创建数组

NumPy 数组是 NumPy 的核心,它提供了多种创建数组的方法。

import numpy as np

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

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

数组索引

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

print(array_1d[0])  # 输出: 1
print(array_2d[0, 1])  # 输出: 2

数组操作

NumPy 提供了大量的数组操作函数。

# 数组求和
print(np.sum(array_2d))  # 输出: 10

# 数组转置
print(np.transpose(array_2d))  # 输出: [[1 3], [2 4]]

高级操作

矩阵运算

NumPy 提供了大量的矩阵运算函数。

# 矩阵乘法
print(np.dot(array_2d, array_2d))  # 输出: [[7 10], [15 22]]

形状操作

NumPy 还支持数组形状的转换。

# 调整数组形状
print(array_2d.shape)  # 输出: (2, 2)
print(array_2d.reshape(4))  # 输出: [1 2 3 4]

扩展阅读

想要了解更多关于 NumPy 的知识,可以参考我们的NumPy 官方文档

NumPy