时间戳

首页 > 解决方案 > 连接矩阵中每个 Numpy 数组的索引

python - 连接矩阵中每个 Numpy 数组的索引

问题描述

所以我有一个 Numpy 数组，里面有一堆 numpy 数组。我想根据它们在各自数组中的位置对它们进行分组。

例如：如果矩阵是：

[[1, 2], [2, 3], [4, 5], [6, 7]]

然后代码应该返回：

[[1, 2, 4, 6], [2, 3, 5, 7]]

这是因为1, 2, 4, 6它们都是它们各自数组中的第一个元素，并且2, 3, 5, 7是它们各自数组中的第二个元素。

任何人都知道一些可以做到这一点的功能。谢谢。

用 Python 回答。

标签： pythonarraysnumpy

解决方案

使用numpy transpose应该可以解决问题：

a = np.array([[1, 2], [2, 3], [4, 5], [6, 7]])
a_t = a.T
print(a_t)
array([[1, 2, 4, 6],
       [2, 3, 5, 7]])

推荐阅读