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。

更多 Pandas 教程

NumPy Logo