代码之家 › 专栏 › 技术社区 › Joan

基于每行选定值的总和对np数组排序

sorting arrays python

Joan · 技术社区 · 2 年前

我有一个2D numpy数组,里面充满了浮点数。

例如:

array([[0.80372444, 0.35468653, 0.9081662 , 0.69995566],
       [0.53712474, 0.90619077, 0.69068265, 0.73794143],
       [0.14056974, 0.34685164, 0.87505744, 0.56927803]])

以下是我尝试的内容:

a = np.array(sorted(a, key = sum))

但这只是对每行的所有值求和,而不是说,只对第二到第六个元素求和。

2 回复 | 直到 2 年前

James 2 年前

你可以使用 take [1,2] 从每行(轴=1)。然后 sum argsort 求和的顺序。这提供了一组行索引,您可以使用这些索引按所需顺序对数组进行切片。

import numpy as np

a = np.array([[0.80372444, 0.35468653, 0.9081662 , 0.69995566],
              [0.53712474, 0.90619077, 0.69068265, 0.73794143],
              [0.14056974, 0.34685164, 0.87505744, 0.56927803]])

a[a.take([1, 2], axis=1).sum(axis=1).argsort()]
# returns:
array([[0.14056974, 0.34685164, 0.87505744, 0.56927803],
       [0.80372444, 0.35468653, 0.9081662 , 0.69995566],
       [0.53712474, 0.90619077, 0.69068265, 0.73794143]])

Scott Hunter 2 年前

key

a = np.array(sorted(d, key = lambda v : sum(v[1:3])))

推荐文章

July · 如何定义数字间隔,然后四舍五入

1 年前

Community wiki · 对象名称前的单下划线和双下划线的含义是什么?

1 年前

Brian Johnson · 为什么在Python中列出字典列表会引发TypeError?[已关闭]

1 年前

user026 · 如何根据特定窗口的平均值(行数)创建新列?

1 年前

Ashok Shrestha · 需要追踪特定的颜色线并获取坐标

1 年前

Nicote Ool · 在FastApi和Vue3中获得422

1 年前

NeoExceptCodeBad · 如果我有很多垂直线,我如何找到它们的边缘?

1 年前

Abdulaziz · 如何对集合内的列表进行排序[重复]

1 年前

user2743931 · 带有src目录的Python setup.py

1 年前

asmgx · 为什么合并数据帧不能按照python中的预期方式工作

1 年前