NumPy 数组操作：从入门到精通

news2024/10/19 7:14:00

引言

随着大数据时代的到来，如何高效地存储、处理大量数据成为了一个亟待解决的问题。传统的Python列表虽然灵活，但在面对大规模数据集时显得力不从心。NumPy正是在这种背景下应运而生，它提供了一种高效的数据结构——数组（Array），能够以更低的空间开销存储相同数量的数据，并且支持向量化运算，极大地提升了数据处理速度。无论是进行科学计算、数据分析还是机器学习模型训练，NumPy都是不可或缺的工具之一。

基础语法介绍

数组创建

numpy.array(): 最常用的数组创建方式，可以将列表或其他序列转换为数组。
numpy.zeros(), numpy.ones(), numpy.empty(): 创建特定形状的数组，分别初始化为0、1或未初始化值。
numpy.arange(), numpy.linspace(), numpy.logspace(): 生成等差数列、等比数列或对数等比数列。

import numpy as np

# 从列表创建数组
a = np.array([1, 2, 3])
print(a)  # 输出: [1 2 3]

# 创建零数组
b = np.zeros((2, 3))
print(b)
# 输出:
# [[0. 0. 0.]
#  [0. 0. 0.]]

# 创建等差数列
c = np.arange(1, 10, 2)
print(c)  # 输出: [1 3 5 7 9]

数组索引与切片

单一元素访问：arr[index]
多维数组索引：arr[row, column]
切片操作：arr[start:end:step]

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

数组运算

算术运算：加(+), 减(-), 乘(*), 除(/), 指数(**), 取模(%)
布尔运算：与(&), 或(|), 非(~)
广播机制：允许不同形状的数组之间进行运算

x = np.array([1, 2, 3])
y = np.array([4, 5, 6])

print(x + y)  # 输出: [5 7 9]
print(x * y)  # 输出: [ 4 10 18]

基础实例

假设我们需要对一个包含温度记录的数组进行处理，将其从摄氏度转换为华氏度。

celsius_temps = np.array([-20, -15, 0, 5, 10, 15, 20])
fahrenheit_temps = celsius_temps * (9 / 5) + 32
print(fahrenheit_temps)
# 输出: [-4.  5. 32. 41. 50. 59. 68.]

进阶实例

接下来，我们尝试使用NumPy处理一个稍微复杂些的问题：给定两个不同长度的数组，如何找到它们之间的交集？

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

intersect = np.intersect1d(a, b)
print(intersect)  # 输出: [4 5]

此外，NumPy还提供了丰富的函数来处理数组的排序、统计分析等功能，例如np.sort()、np.mean()、np.median()等，可以帮助我们更好地理解数据分布特征。

实战案例

在图像处理领域，NumPy经常被用来读取、编辑图像文件。下面是一个简单的例子，演示如何利用NumPy读取一张图片，并将其转换为灰度图。

from PIL import Image
import numpy as np

img = Image.open('example.jpg')
img_array = np.array(img)

gray_img_array = np.dot(img_array[...,:3], [0.299, 0.587, 0.114]).astype(np.uint8)
gray_img = Image.fromarray(gray_img_array)
gray_img.save('gray_example.jpg')