在NumPy数组中对字典进行排序：获取并按值排序内嵌字典的实用指南-Python教程-PHP中文网

在NumPy数组中对字典进行排序：获取并按值排序内嵌字典的实用指南

聖光之護

发布： 2025-09-28 13:33:25

原创

480人浏览过

在NumPy数组中对字典进行排序：获取并按值排序内嵌字典的实用指南

本教程详细介绍了如何在NumPy数组中对内嵌的Python字典进行排序。当一个字典作为NumPy数组的唯一元素时，我们将学习如何使用.item()方法提取该字典，并利用Python内置的sorted()函数结合lambda表达式，根据字典的值（例如，城市伤亡总数）进行降序排序，最终生成一个按需排列的新字典。

理解问题：NumPy数组中的字典结构

在数据处理过程中，有时我们会将python字典作为numpy数组的一个元素进行存储。例如，在统计城市伤亡总数时，我们可能会得到一个形如 {'城市名': 伤亡总数} 的字典。当我们将这个字典直接传递给 np.array() 函数时，numpy并不会将其内部元素进行向量化处理，而是将其作为一个单一的python对象存储在数组中。

考虑以下场景，我们已经从CSV文件中提取并计算了印度各城市的总伤亡人数，结果存储在一个字典 city_dict 中：

import csv
import numpy as np

# 假设 city_dict 已经通过处理 terrorismData.csv 文件生成
# 示例数据（实际数据可能更多）：
city_dict = {
    'New Delhi': 2095, 'Samastipur': 4, 'Bombay': 210, 'Imphal': 603, 'Aizawl': 2,
    'Amapur': 2, 'Raisikah': 1, 'Champhai': 1, 'Jamshedpur': 32, 'Chennai': 366,
    'Chiaplant': 1, 'Tindol': 7, 'Calcutta': 57, 'Tirupattur': 6, 'Gauhati': 112,
    'Jorhat': 3, 'Massad': 1, 'Chandigarh': 333, 'Jodhpur': 2, 'Amritsar': 768,
    'Tipaimukh': 6, 'Guwahati': 822, 'Harchowal': 1, 'Mothan Wala': 2, 'Qadian': 7,
    'Baloda Bazar': 10
}

np_city = np.array(city_dict)
print(np_city)

登录后复制

输出结果会是类似 array({'New Delhi': 2095, ...}, dtype=object) 的形式。这表明 np_city 是一个只包含一个元素的NumPy数组，而这个元素就是我们原始的 city_dict 字典。直接对 np_city 进行字典操作是行不通的，因为 np_city 本身不是一个字典。

提取内嵌字典

要对这个内嵌的字典进行操作，我们首先需要将其从NumPy数组中提取出来。对于只包含一个元素的NumPy数组（特别是当该元素是Python对象时），可以使用 .item() 方法来获取该元素。

# 提取内嵌的字典对象
actual_dict = np_city.item()
print(type(actual_dict))
print(actual_dict)

登录后复制

此时，actual_dict 就是我们原始的 city_dict，一个标准的Python字典。

对字典按值进行排序

一旦我们成功提取了字典，就可以使用Python内置的 sorted() 函数对其进行排序。要根据字典的值进行降序排序，我们需要结合 dict.items() 方法、lambda 表达式和 reverse=True 参数。

怪兽AI数字人

数字人短视频创作，数字人直播，实时驱动数字人

查看详情

dict.items(): 该方法返回一个包含字典所有(键, 值)对的视图对象。
sorted(): 这个函数可以对任何可迭代对象进行排序。
key=lambda item: item[1]: 这是排序的关键。lambda 函数定义了一个匿名函数，它接收一个 item（即一个 (键, 值) 对），并返回 item[1]（即值）。sorted() 函数将根据这个返回的值来决定排序顺序。
reverse=True: 这个参数指定了排序的方向，True 表示降序排列。

将这些组合起来，并使用字典推导式 (dictionary comprehension) 将排序后的键值对重新构建成一个新的字典：

# 对字典按值进行降序排序
sorted_city_dict = {
    key: value for key, value in sorted(
        actual_dict.items(),
        key=lambda item: item[1],
        reverse=True
    )
}

print(sorted_city_dict)

登录后复制

完整示例代码

以下是整合了数据准备、字典提取和排序的完整代码：

import csv
import numpy as np

# 模拟从CSV文件处理数据并生成字典的过程
# 实际应用中，city_dict 会通过读取文件动态生成
city_dict = {
    'New Delhi': 2095, 'Samastipur': 4, 'Bombay': 210, 'Imphal': 603, 'Aizawl': 2,
    'Amapur': 2, 'Raisikah': 1, 'Champhai': 1, 'Jamshedpur': 32, 'Chennai': 366,
    'Chiaplant': 1, 'Tindol': 7, 'Calcutta': 57, 'Tirupattur': 6, 'Gauhati': 112,
    'Jorhat': 3, 'Massad': 1, 'Chandigarh': 333, 'Jodhpur': 2, 'Amritsar': 768,
    'Tipaimukh': 6, 'Guwahati': 822, 'Harchowal': 1, 'Mothan Wala': 2, 'Qadian': 7,
    'Baloda Bazar': 10
}

# 将字典放入NumPy数组
np_city = np.array(city_dict)
print("原始NumPy数组内容：")
print(np_city)
print("-" * 30)

# 1. 提取NumPy数组中的字典对象
actual_dict = np_city.item()
print("提取出的字典对象：")
print(actual_dict)
print("-" * 30)

# 2. 对字典按值进行降序排序，并创建新的有序字典
sorted_city_dict = {
    key: value for key, value in sorted(
        actual_dict.items(),
        key=lambda item: item[1],
        reverse=True
    )
}

print("按伤亡总数降序排序后的城市字典：")
print(sorted_city_dict)

# 如果需要获取前N个城市，可以对排序后的字典进行切片（Python 3.7+ 字典保持插入顺序）
top_5_cities = dict(list(sorted_city_dict.items())[:5])
print("\n伤亡总数前5的城市：")
print(top_5_cities)

登录后复制

输出示例：

原始NumPy数组内容：
{'New Delhi': 2095, 'Samastipur': 4, 'Bombay': 210, 'Imphal': 603, 'Aizawl': 2, 'Amapur': 2, 'Raisikah': 1, 'Champhai': 1, 'Jamshedpur': 32, 'Chennai': 366, 'Chiaplant': 1, 'Tindol': 7, 'Calcutta': 57, 'Tirupattur': 6, 'Gauhati': 112, 'Jorhat': 3, 'Massad': 1, 'Chandigarh': 333, 'Jodhpur': 2, 'Amritsar': 768, 'Tipaimukh': 6, 'Guwahati': 822, 'Harchowal': 1, 'Mothan Wala': 2, 'Qadian': 7, 'Baloda Bazar': 10}
------------------------------
提取出的字典对象：
{'New Delhi': 2095, 'Samastipur': 4, 'Bombay': 210, 'Imphal': 603, 'Aizawl': 2, 'Amapur': 2, 'Raisikah': 1, 'Champhai': 1, 'Jamshedpur': 32, 'Chennai': 366, 'Chiaplant': 1, 'Tindol': 7, 'Calcutta': 57, 'Tirupattur': 6, 'Gauhati': 112, 'Jorhat': 3, 'Massad': 1, 'Chandigarh': 333, 'Jodhpur': 2, 'Amritsar': 768, 'Tipaimukh': 6, 'Guwahati': 822, 'Harchowal': 1, 'Mothan Wala': 2, 'Qadian': 7, 'Baloda Bazar': 10}
------------------------------
按伤亡总数降序排序后的城市字典：
{'New Delhi': 2095, 'Guwahati': 822, 'Amritsar': 768, 'Imphal': 603, 'Chennai': 366, 'Chandigarh': 333, 'Bombay': 210, 'Gauhati': 112, 'Calcutta': 57, 'Jamshedpur': 32, 'Baloda Bazar': 10, 'Tindol': 7, 'Qadian': 7, 'Tirupattur': 6, 'Tipaimukh': 6, 'Samastipur': 4, 'Jorhat': 3, 'Aizawl': 2, 'Amapur': 2, 'Jodhpur': 2, 'Mothan Wala': 2, 'Raisikah': 1, 'Champhai': 1, 'Chiaplant': 1, 'Massad': 1, 'Harchowal': 1}

伤亡总数前5的城市：
{'New Delhi': 2095, 'Guwahati': 822, 'Amritsar': 768, 'Imphal': 603, 'Chennai': 366}

登录后复制

注意事项与总结

NumPy数组与Python字典的存储方式：当一个Python字典被 np.array() 包裹时，如果该字典是唯一的元素，NumPy会将其作为一个 object 类型的元素存储，而不是尝试解析其内部结构。因此，不能直接对 np_city 进行字典操作。
.item() 方法：这个方法对于从只包含单个元素的NumPy数组中提取该元素非常有用。
字典排序：Python字典本身是无序的（在Python 3.7+中保持插入顺序，但并非按键或值自动排序）。sorted() 函数返回的是一个排序后的键值对列表。为了得到一个保持排序顺序的字典，我们需要使用字典推导式将其重新构建。如果需要更严格的有序字典行为（例如，在旧版Python中），可以使用 collections.OrderedDict。
性能考量：对于非常大的字典，这种方法是有效的。如果数据量极其庞大且需要更复杂的数值计算或结构化操作，考虑使用 Pandas DataFrame 可能会提供更高的效率和更丰富的功能。Pandas DataFrame 可以直接从字典创建，并提供强大的排序、过滤和分析能力。

通过理解NumPy处理Python对象数组的机制以及Python字典的排序方法，我们可以有效地解决在NumPy数组中对内嵌字典进行排序的需求。

以上就是在NumPy数组中对字典进行排序：获取并按值排序内嵌字典的实用指南的详细内容，更多请关注php中文网其它相关文章！