Python array模块数组怎么添加元素？

一、操作前提：必须先理解 array 的类型码约束

在直接调用添加方法之前，必须明确 array 与 Python 普通 list 的根本区别。Python 的 list 是异构容器，列表中存放的是对象的指针引用，因此可以随意混放整数、浮点数甚至字符串。而 array.array 是对 C 语言原生数组的轻量级封装，要求数组内所有元素必须是同质数据类型（Homogeneous）。

当实例化一个数组时，第一个参数必须是类型码（Type Code）。常见类型码及其限制如下表所示：

二、添加单个元素：append() 与 insert()

向 array 中添加单个值时，最常用的是 append() 与 insert() 方法。

1. 使用 append(x) 向末尾追加元素

append(x) 是最直接的方法，它将单个元素追加到现有数组的末尾。该操作的时间复杂度平摊为 O(1)，执行效率最高。

import array

# 创建一个存储有符号整数('i')的数组
numbers = array.array('i', [10, 20, 30])

# 向数组末尾添加单个整数 40
numbers.append(40)

print("当前数组内容:", numbers)
print("数组元素列表:", numbers.tolist())

2. 使用 insert(i, x) 在指定索引处插入元素

如果需要将元素放到数组的首部或特定位置，可以使用 insert(i, x)。其中参数 i 代表目标插入位置的索引编号，x 为要插入的值。

需要注意，数组底层是连续物理内存，如果在头部或中间插入元素，后续的所有元素都需要在内存中依次向后移动一位，因此时间复杂度为 O(n)。

import array

numbers = array.array('i', [10, 20, 30, 40])

# 在索引 0（数组头部）插入数字 5
numbers.insert(0, 5)

# 在索引 2 处插入数字 15
numbers.insert(2, 15)

print("插入后数组内容:", numbers.tolist())

三、批量添加元素：extend() 与 fromlist()

当需要一次性追加多个元素时，循环调用 append() 会产生显著的函数调用开销。此时应使用专用的批量添加方法。

1. 使用 extend(iterable) 追加可迭代对象

extend() 接收任何包含合法数据的可迭代对象（如 list、tuple、range 或另一个相同类型码的 array），并将其中所有元素按顺序追加到数组末尾。

import array

# 创建浮点数('f')数组
float_arr = array.array('f', [1.5, 2.5])

# 批量追加元组中的浮点数据
float_arr.extend((3.5, 4.5, 5.5))

print("extend 后的浮点数组:", float_arr.tolist())

2. 使用 fromlist(list) 从列表导入

fromlist(list) 是 array 模块特有的接口，专门用于将标准 Python 列表中的所有项追加到数组中。它的运行行为类似于对列表执行 extend()，但限制参数必须是标准 list。

import array

int_arr = array.array('i', [1, 2, 3])
data_list = [4, 5, 6]

# 从列表中快速导入并追加
int_arr.fromlist(data_list)

print("fromlist 后的整数数组:", int_arr.tolist())

3. 使用加法运算符拼接两个数组

除了调用实例方法外，还可以使用 + 或 += 运算符将两个 array 拼接在一起。但必须保证两个数组的类型码完全一致，否则 Python 会拒绝合并并报错。

import array

arr_a = array.array('i', [10, 20])
arr_b = array.array('i', [30, 40])

# 使用 += 原地追加
arr_a += arr_b

print("拼接后的 arr_a:", arr_a.tolist())

四、关键转折与常见报错：TypeError 与 OverflowError 排查

在实际使用中，初学者向 array 添加元素时最容易遭遇两类致命异常：类型不匹配（TypeError）与数值溢出（OverflowError）。

1. 浮点数/字符串无法直接插入整型数组 (TypeError)

在 list 中可以任意混合数据，但在 array('i') 中添加浮点数或字符串会立刻抛出异常：

import array

arr = array.array('i', [1, 2, 3])

# 错误尝试：向整型数组追加浮点数 4.5
try:
    arr.append(4.5)
except TypeError as e:
    print("捕获到 TypeError:", e)

2. 数值超出类型码范围 (OverflowError)

如果数组使用的是小尺寸整数类型码（例如 'b' 代表有符号单字节整数，数值区间为 -128 ~ 127），一旦尝试添加 128 或更大值，就会触发溢出报错：

import array

# 创建单字节有符号数组（-128 至 127）
byte_arr = array.array('b', [10, 20])

try:
    byte_arr.append(200)  # 超出 127
except OverflowError as e:
    print("捕获到 OverflowError:", e)

五、方法选型与性能建议

不同的添加方式在执行性能与内存开销上存在明显差异。在实际开发中，应结合数据来源与操作频次做出合理选择：

六、总结与关键清单

向 Python array 模块数组添加元素的操作非常明确，但由于其底层对应 C 语言的原生紧凑存储，务必牢记以下三条铁律：

• 添加单一值：首选 append()；仅在必须精准控制元素位置时才使用 insert()，以避免不必要的数组元素全量右移开销。
• 批量添加：当源数据为标准 list 时，使用 fromlist()；对于元组、生成器或另一个同类型 array，使用 extend() 或 += 操作符。
• 防御性校验：添加前确保数值符合类型码定义，避免浮点数直插整型数组导致 TypeError，以及极端数值超出字节上下限导致 OverflowError。