NumPy 基础

什么是 NumPy？

NumPy (Numerical Python) 是 Python 科学计算的核心库，提供了高性能的多维数组对象以及处理这些数组的工具。

pip install numpy

import numpy as np

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

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

索引操作类似于 Python 列表，但功能更强大。

arr = np.array([1, 2, 3, 4])
print(arr[0]) # 输出 1
print(arr[1:3]) # 输出 [2, 3]

print(arr_2d.shape) # 输出 (2, 3)

NumPy 支持对数组进行逐元素的数学运算。

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
print(a + b) # 输出 [5, 7, 9]
print(a * 2) # 输出 [2, 4, 6]

zeros = np.zeros((2, 3)) # 全 0 数组
ones = np.ones((2, 3))   # 全 1 数组
rng = np.arange(10)      # 类似于 range() -> [0, 1, ..., 9]