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 中做到这一点吗？

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