用魔法帮你把一维数组和二维数组相互转换，它就是Python array!

2023-10-16 07:54:52

掌握Python array：轻松转换一维与二维数组

对于Python开发人员来说，处理数据是家常便饭。然而，有时候我们可能遇到这样的难题：收到的数据不是我们期望的一维格式，而是二维数组。别担心！Python array模块就像一位神奇的魔术师，可以轻松帮我们解决这个问题。

什么是Python array？

Python array是一个内置模块，用于创建和操作一维数组。一维数组与列表类似，包含一系列相同数据类型的元素。使用array.array()函数，我们可以轻松创建一维数组。

如何将一维数组转换为二维数组

Python array提供了reshape()函数，可以将一维数组转换为二维数组。reshape()函数有两个参数：行数和列数。例如，我们要将一维数组[1, 2, 3, 4, 5, 6, 7, 8, 9]转换为一个3行3列的二维数组，可以使用以下代码：

import array

array1 = array.array('i', [1, 2, 3, 4, 5, 6, 7, 8, 9])
array2 = array1.reshape(3, 3)

如何将二维数组转换为一维数组

与转换一维数组到二维数组类似，我们可以使用flatten()函数将二维数组转换为一维数组。flatten()函数会将二维数组中的元素依次展开成一个一维数组。例如，要将二维数组[[1, 2, 3], [4, 5, 6], [7, 8, 9]]转换为一维数组，可以使用以下代码：

import array

array2 = array.array('i', [[1, 2, 3], [4, 5, 6], [7, 8, 9]])
array1 = array2.flatten()

示例代码

了解了Python array的基本用法后，让我们通过一个代码示例来加深理解：

import array

# 创建一维数组
array1 = array.array('i', [1, 2, 3, 4, 5, 6, 7, 8, 9])

# 将一维数组转换为二维数组
array2 = array1.reshape(3, 3)

# 输出二维数组
print(array2)

# 将二维数组转换为一维数组
array1 = array2.flatten()

# 输出一维数组
print(array1)