Python 如何从字典中删除一个 NaN 键

NaN（Not a Number）为IEEE浮点数标准中的特殊值，表示无法表示或计算的数值。当我们需要从一个字典中删除NaN键时，可能会遇到一些问题。本文将介绍如何从字典中删除NaN键。

使用 pandas 库

如果我们的字典是由DataFrame转换而来，我们可以使用pandas库来删除NaN键。下面的示例演示了如何删除字典df中的所有NaN键。

import pandas as pd

# 初始化字典
df = pd.DataFrame({'a': [1, 2, 3], 'b': [4, float('nan'), 6], 'c': [7, 8, 9]})
print('原始字典：', df.to_dict())

# 删除NaN键
df.dropna(axis=1, inplace=True)
print('删除NaN键后的字典：', df.to_dict())

输出：

原始字典： {'a': {0: 1, 1: 2, 2: 3}, 'b': {0: 4.0, 1: nan, 2: 6.0}, 'c': {0: 7, 1: 8, 2: 9}}
删除NaN键后的字典： {'a': {0: 1, 1: 2, 2: 3}, 'c': {0: 7, 1: 8, 2: 9}}

使用列表推导式

如果我们的字典是由普通列表转换而来，我们可以使用列表推导式来删除NaN键。下面的示例演示了如何删除字典d中的所有NaN键。

# 初始化字典
d = {'a': 1, 'b': float('nan'), 'c': 3}
print('原始字典：', d)

# 删除NaN键
d = {k: v for k, v in d.items() if v == v}
print('删除NaN键后的字典：', d)

输出：

原始字典： {'a': 1, 'b': nan, 'c': 3}
删除NaN键后的字典： {'a': 1, 'c': 3}

列表推导式的核心是{k: v for k, v in d.items() if v == v}。这行代码遍历字典d的每一个键值对，当值不等于NaN时，生成一个新的键值对。生成的字典中已经不包含NaN键了。

总结

本文介绍了两种从字典中删除NaN键的方法：使用pandas库和使用列表推导式。这两种方法都比较简单，可以满足大多数应用场景的需求。