如何在python 3中将2D数组转换为元组？

Question

类型转换

我有一个大小为667000 * 3的numpy数组，我想将其转换为667000 * 3元组。

在较小的尺寸中，这就像将arr转换为t，其中：

arr= [[1,2,3],[4,5,6],[7,8,9],[10,11,12]]

t= ((1,2,3),(4,5,6),(7,8,9),(10,11,12))

我尝试过：

t = tuple((map(tuple, sub)) for sub in arr)   

但是没有用。

您能帮我在python 3中怎么做吗？

Answer 1

您无需遍历sub，只需将每个子列表包装在一个元组中，然后将结果包装在一个元组中，例如：

tuple(map(tuple, arr))

例如：

>>> arr = [[1,2,3],[4,5,6],[7,8,9],[10,11,12]]
>>> tuple(map(tuple, arr))
((1, 2, 3), (4, 5, 6), (7, 8, 9), (10, 11, 12))

此处map将产生一个生成器，该生成器将为每个子列表（如[1, 2, 3]）将其转换为元组（如(1, 2, 3)）。外部tuple(..)构造函数会将这个生成器的元素包装在一个元组中。

根据实验，转换667000×3矩阵是可行的。当我在np.arange(667000*3)和np.random.rand(667000, 3)上运行它时，它需要0.512秒：

>>> arr = np.random.rand(667000,3)
>>> timeit.timeit(lambda: tuple(map(tuple, arr)), number=10)
5.120870679005748
>>> arr = np.arange(667000*3).reshape(-1, 3)
>>> timeit.timeit(lambda: tuple(map(tuple, arr)), number=10)
5.109966446005274

Answer 2

一个简单的迭代解决方案是使用生成器表达式：

tuple(tuple(i) for i in arr)
# ((1, 2, 3), (4, 5, 6), (7, 8, 9), (10, 11, 12))