python - 如何在 python 3 中将二维数组转换为元组？

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中怎么做吗？

标签： pythonarraystype-conversiontuples

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将生成一个生成器，该生成器将为每个子列表（like [1, 2, 3]）将其转换为元组（like (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