numpy:展平2D数组的某些列

艾伦

假设我有一个如下的numpy数组

a = np.asarray([[1,2,3],[1,4,3],[2,5,4],[2,7,5]])

array([[1, 2, 3],
       [1, 4, 3],
       [2, 5, 4],
       [2, 7, 5]])

如何为第1列中的每个唯一元素展平第2列和第3列,如下所示:

array([[1, 2, 3, 4, 3],
       [2, 5, 4, 7, 5],])

谢谢您的帮助。

智慧

使用列表理解的另一种选择:

np.array([np.insert(a[a[:,0] == k, 1:].flatten(), 0, k) for k in np.unique(a[:,0])])

# array([[1, 2, 3, 4, 3],
#        [2, 5, 4, 7, 5]])

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章

来自分类Dev

展平2D数组1D和

来自分类Dev

将6D数组以numpy展平为2D的惯用方式,同时保留汉密尔顿结构

来自分类Dev

如何展平第二轴上具有不同长度的2d numpy数组?

来自分类Dev

如何仅展平numpy数组的某些维

来自分类Dev

将数组从2d展平到1d

来自分类Dev

展平JavaScript 2D数组是否值得提高性能?

来自分类Dev

如何防止Elasticsearch在包含“字段”的查询中展平2D数组

来自分类Dev

展平NumPy数组列表?

来自分类Dev

numpy 2d数组操作,保持某些位置

来自分类Dev

如何在不使用numpy的情况下将2D列表展平为1D?

来自分类Dev

将包含对象的3D数组展平为2D,通过其参数删除重复的对象

来自分类Dev

Python在numpy数组中展平数组

来自分类Dev

可以重建2d矩阵,其中已将遮罩与numpy一起使用,并且已展平

来自分类Dev

试图在numpy中展平数组

来自分类Dev

numpy 2d和1d加法平

来自分类Dev

将包含numpy数组的元组展平为numpy数组

来自分类Dev

在2D numpy数组的末尾添加一列

来自分类Dev

按多列对numpy 2d数组进行排序

来自分类Dev

按列索引排序numpy 2d数组

来自分类Dev

在python中将3d列表展平为2d

来自分类Dev

我如何设置2D数组中某些元素的值(使用numpy在列表中)?

来自分类Dev

Python numpy 2D数组总和在某些索引上

来自分类Dev

在TensorFlow中展平包含向量的2D张量的最佳方法?

来自分类Dev

展平熊猫数组

来自分类Dev

展平多维数组

来自分类Dev

递归展平数组

来自分类Dev

展平多维数组

来自分类Dev

展平多维数组

来自分类Dev

展平任意数组