在python中使用numpy在一列中标识具有相同值的向量

发布时间：2020-12-20 11:42:55 所属栏目：Python 来源：网络整理

导读：我有一个大的2d矢量数组.我想根据矢量的元素或维度之一将这个数组拆分成几个数组.如果此列中的值连续相同,我希望收到一个这样的小数组.例如,考虑第三维或列： orig = np.array([[1,2,3],[3,4,[5,6,4],[7,8,[9,[8,7,[6,5,3]]) 我想变成三个由1,2和3,5和6,7行

我有一个大的2d矢量数组.我想根据矢量的元素或维度之一将这个数组拆分成几个数组.如果此列中的值连续相同,我希望收到一个这样的小数组.例如,考虑第三维或列：

orig = np.array([[1,2,3],[3,4,[5,6,4],[7,8,[9,[8,7,[6,5,3]])

我想变成三个由1,2和3,5和6,7行组成的数组：

>>> a
array([[1,3]])

>>> b
array([[5,4]])

>>> c
array([[8,3]])

我是python和numpy的新手.任何帮助将不胜感激.

问候
垫

编辑：我重新格式化了数组以澄清问题

解决方法

使用 np.split：

>>> a,b,c = np.split(orig,np.where(orig[:-1,2] != orig[1:,2])[0]+1)

>>> a
array([[1,[1,3]])
>>> b
array([[1,4]])
>>> c
array([[1,3]])

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!