基于给定的属性索引对表格进行分类的 Python 程序

假设我们有一个 2 维数组，包含运动员信息。该信息包括排名、年龄、身高。每行包含不同运动员的信息。我们还有一个数字 k。我们必须根据第 k 个属性对数据进行排序。

因此，如果输入类似于

而且 k = 1。

那么输出将是

[[1, 25, 190], [4, 26, 175], [3, 33, 185], [2, 35, 180], [5, 35, 180]]

为了解决这个问题，我们将执行以下步骤——

• 为名为 info 的 2D 数组调用 sort() 函数

• 定义一个基于第 k 个参数进行排序的功能，并将其传递给 sort() 函数的 key 参数。

范例

让我们看看以下实现以获得更好的理解

def solve(info, k):
   info.sort(key = lambda x: x[k])
   return info

info = [[1, 25, 190],[2, 35, 180],[3, 33, 185],[4, 26, 175],[5, 35, 180]]
k = 1
print(solve(info, k))

输入

[[1, 25, 190],[2, 35, 180],[3, 33, 185],[4, 26, 175],[5, 35, 180]], 1

输出

[[1, 25, 190], [4, 26, 175], [3, 33, 185], [2, 35, 180], [5, 35, 180]]

Arnab Chakraborty

更新于：12-Oct-2021

371 次浏览

开启你的 职业

通过完成课程来获得认证

入门