NumPy 是 Python 中一个强大的科学计算库,它提供了大量用于数值计算的函数和方法。以下是一些 NumPy 的基本概念和用法。
NumPy 简介
NumPy 提供了一个强大的 N 维数组对象,以及一系列用于快速操作数组的函数。
安装 NumPy
首先,您需要安装 NumPy。您可以使用 pip 来安装:
pip install numpy
创建 NumPy 数组
NumPy 数组是 NumPy 的核心。以下是创建 NumPy 数组的一些基本方法:
使用
numpy.array()
函数创建数组:import numpy as np arr = np.array([1, 2, 3, 4]) print(arr)
使用
numpy.zeros()
或numpy.ones()
创建全零或全一的数组:zeros_arr = np.zeros((3, 3)) ones_arr = np.ones((2, 2))
数组操作
NumPy 提供了大量的数组操作方法,以下是一些常用的操作:
数组索引:
arr[0] # 获取第一个元素 arr[1:3] # 获取第二个和第三个元素
数组切片:
arr[1:3:2] # 获取从第二个到第三个,步长为2的元素
数组形状:
arr.shape # 返回数组的形状
NumPy 矩阵运算
NumPy 提供了矩阵运算的功能,例如矩阵乘法:
import numpy as np
A = np.array([[1, 2], [3, 4]])
B = np.array([[2, 0], [1, 3]])
result = np.dot(A, B)
print(result)
NumPy 与 Pandas
NumPy 是 Pandas 库的基础,如果您对数据分析和处理感兴趣,建议您学习 Pandas。
NumPy Logo