拼接并排序numpy嵌套邮编列表时报AxisError和TypeError如何解决

错误原因分析 #

• 第一个AxisError：执行np.concatenate(b)后得到的a是0维标量而非一维数组，说明用于拼接的嵌套列表b结构不符合要求，大概率是b内部的子数组维度不统一、或部分子数组为空，导致拼接失败返回了0维对象。
• 第二个TypeError：数组中混合了两类数据类型：正常邮编为字符串类型，而CSV中空单元格被numpy自动转为NaN（属于float类型），排序时字符串和浮点数无法直接比较导致报错。

解决步骤 #

1. 先修正拼接逻辑，确保得到一维数组且统一数据类型

# 先过滤b中的空子数组，再执行拼接
valid_b = [arr for arr in b if arr.ndim > 0 and len(arr) > 0]
a = np.concatenate(valid_b).astype(str) # 统一转为字符串类型，NaN会被转为字符串'nan'

2. 过滤空值后执行排序

# 过滤无效空值后按字母顺序排序
valid_a = a[a != 'nan']
sorted_postcodes = np.sort(valid_a)
print(sorted_postcodes)

# 如果需要保留空值且让空值排在末尾，可使用稳定排序
sorted_postcodes = np.sort(a, kind='mergesort')

3. 兼容方案：转为Python原生列表处理，适配性更强

postcode_list = a.tolist()
# 过滤空值后排序
valid_list = [x for x in postcode_list if x != 'nan']
valid_list.sort()
print(valid_list)

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