Question

我是Python新手。我使用的是Python 2.7。我想创建一个2D数组，我知道如何使用列表。但是使用列表数据很大。为了节省内存，我想使用数组而不是列表。这是受到了启发 '使用array.array（'l'）而不是回答Python memory usage? loading large dictionaries in memory

中给出的（整数）值列表

这种方法可以用于2D阵列吗？非常感谢。

Answer 1

您无法真正创建2d array.array()，因为它们的元素仅限于以下类型：字符，整数和浮点数。相反，您可以将数据存储在常规的一维数组中，并通过一些辅助函数访问它。

以下是我想要描述的内容的插图：

from array import array

INFO_SIZE = 3  # number of entries used to store info at beginning of array
WIDTH, HEIGHT = 1000, 1000  # array dimensions
array2d = array('l', (0 for _ in xrange(INFO_SIZE + WIDTH*HEIGHT)))
array2d[:INFO_SIZE] = array('l', (INFO_SIZE, WIDTH, HEIGHT))  # save array info

def get_elem(two_d_array, i, j):
    info_size = two_d_array[0]
    width = two_d_array[1]
    return two_d_array[info_size + j*width + i]

def set_elem(two_d_array, i, j, value):
    info_size = two_d_array[0]
    width = two_d_array[1]
    two_d_array[info_size + j*width + i] = value

import sys
print format(sys.getsizeof(array2d), ",d")  # 4,091,896

print get_elem(array2d, 999, 999)  # 0
set_elem(array2d, 999, 999, 42)
print get_elem(array2d, 999, 999)  # 42

正如您所看到的，array2d的大小（相对而言）仅比数据本身的大小（在这种情况下为4,000,000字节）稍微多一些。您可以完全省去这些功能，只需在线进行偏移计算，以避免在每次访问时调用函数的开销。另一方面，如果这不是一个大问题，你可以更进一步，将所有逻辑封装在一个通用的class Array2D中。

Answer 2

您提到的问题是关于词典，而不是数组。无论如何，你可以这样做，这会创建一个list array个4字节整数，初始化为零，实际上是2D array：

from array import array

width, height = 1000, 1000
array2d = [array('l', (0 for _ in xrange(width))) for _ in xrange(height)]

array2d[999][999] = 42

Answer 3

在python数组中是列表。

另一个问题的记忆优势来自不使用字典。

一般情况下，从“从列表到二维数组”移动时，您将看不到内存节省。

给我一份你的数据样本，我会更新我的答案。

可以使用array.array（）来定义2d数组吗？

3 个答案: