PyCharm中sorted(np_array.tolist())出现类型警告，求代码优化方案

问题背景 #

两段逻辑完全等效的代码：

代码块1（纯Python列表） #

array_1 = [[12, 15], [10, 1], [5, 13]]

print(array_1)
""" output:
[[12, 15], [10, 1], [5, 13]]
"""

print(sorted(array_1))
""" output:
[[5, 13], [10, 1], [12, 15]]
"""

代码块2（NumPy数组转列表） #

import numpy as np

np_array_1 = np.array([[12, 15], [10, 1], [5, 13]])

print(np_array_1)
""" output:
[[12 15]
 [10  1]
 [ 5 13]]
"""

array_1 = np_array_1.tolist()

print(array_1)
""" output:
[[12, 15], [10, 1], [5, 13]]
"""

print(sorted(array_1))
""" output:
[[5, 13], [10, 1], [12, 15]]
"""

直接使用sorted(np_array_1.tolist())时，PyCharm会弹出类型警告：

Expected type 'Iterable[SupportsLessThan | Any]' (matched generic type 'Iterable[SupportsLessThanT]'), got 'object' instead

代码实际运行无异常，但希望消除该警告以获得整洁的代码。

解决方案 #

方案1：显式指定排序key #

通过给sorted添加key参数，让IDE明确元素的比较逻辑，消除类型推断的模糊性：

import numpy as np

np_array_1 = np.array([[12, 15], [10, 1], [5, 13]])
sorted_list = sorted(np_array_1.tolist(), key=lambda x: x)

方案2：添加类型注解 #

给tolist()的结果标注明确的类型，让PyCharm识别出这是包含可比较元素的嵌套列表：

import numpy as np
from typing import List

np_array_1 = np.array([[12, 15], [10, 1], [5, 13]])
array_1: List[List[int]] = np_array_1.tolist()
sorted_list = sorted(array_1)

方案3：使用NumPy内置排序（替代方案） #

如果不需要转为Python列表，直接用NumPy原生排序方法可以完全规避类型警告：

import numpy as np

np_array_1 = np.array([[12, 15], [10, 1], [5, 13]])
# 按行首元素排序，效果和原生sorted一致
sorted_np_array = np.sort(np_array_1, axis=0)
# 如需转回Python列表
sorted_list = sorted_np_array.tolist()

原理说明 #

PyCharm的静态类型检查无法自动推断np.ndarray.tolist()返回的嵌套列表元素具体类型，默认标记为object类型，而sorted要求传入的可迭代对象元素需支持比较操作。通过显式指定key或添加类型注解，能让IDE明确元素的可比较属性，从而消除警告。

内容的提问来源于stack exchange，提问作者Riccardo Volpe