python 对列表排序并返回索引

原创

mob64ca12edad02 2024-03-28 04:59:15 ©著作权

文章标签 Python 饼状图 python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12edad02的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python对列表排序并返回索引

在Python编程中，经常会遇到需要对列表进行排序的情况。有时候我们不仅需要对列表进行排序，还需要获取排序后的元素在原列表中的位置索引。Python提供了一种简便的方法来实现这个需求，即使用sorted()函数。

sorted()函数介绍

sorted()函数是Python内置函数，用于对可迭代对象（如列表、元组等）进行排序操作。sorted()函数的基本语法如下：

sorted(iterable, key=None, reverse=False)

iterable：需要排序的可迭代对象。
key：用于指定排序时的比较函数。
reverse：是否降序排序，默认为升序排序。

返回索引的需求

当我们使用sorted()函数对列表进行排序时，通常会直接得到排序后的结果，但有时候我们也需要知道排序后元素在原列表中的位置索引。这时，可以借助Python的enumerate()函数来实现。

enumerate()函数是Python内置函数，用于将一个可遍历的数据对象组合为一个索引序列，同时列出数据和数据下标。下面是enumerate()函数的基本语法：

enumerate(iterable, start=0)

iterable：待枚举的可迭代对象。
start：枚举的起始值，默认为0。

结合sorted()函数和enumerate()函数，我们可以实现对列表排序并返回排序后元素在原列表中的索引。

示例代码

假设我们有一个包含学生成绩的列表scores，现在我们需要对学生成绩进行排序，并返回排序后的学生成绩在原列表中的索引。

# 原始成绩列表
scores = [85, 92, 78, 90, 88]

# 对成绩列表进行排序并返回索引
sorted_index = sorted(range(len(scores)), key=lambda x: scores[x])

# 输出排序后的成绩及其在原列表中的索引
for index in sorted_index:
    print(f"成绩：{scores[index]}，索引：{index}")

上述代码中，我们首先使用sorted()函数和range()函数对成绩列表进行排序并返回索引，然后通过遍历排序后的索引列表，输出排序后的成绩及其在原列表中的索引。

实际案例分析

为了更直观地展示排序后元素在原列表中的索引，我们可以通过绘制饼状图来呈现成绩分布情况。以下是一个使用matplotlib库来绘制饼状图的实际案例：

import matplotlib.pyplot as plt

# 原始成绩列表
scores = [85, 92, 78, 90, 88]

# 绘制饼状图
labels = [f"学生{i+1}" for i in range(len(scores))]
plt.pie(scores, labels=labels, autopct="%1.1f%%", startangle=140)
plt.axis("equal")
plt.title("学生成绩分布")
plt.show()

通过绘制的饼状图，我们可以直观地了解学生成绩的分布情况，同时结合排序后的索引信息，更好地理解学生成绩的排序过程。