空ndarray与空列表在f-string内比较行为差异及相关疑问

测试代码与输出 #

测试代码 #

import numpy as np

print(f"{np.empty(0)} :: {type(np.empty(0))}")
print(f"{[]} :: {type([])}")

print(f"internal -> {np.empty(0) == []} :: {type(np.empty == [])}")

external = np.empty(0) == []
print(f"external -> {external} :: {type(external)}")

执行输出 #

[] :: <class 'numpy.ndarray'>
[] :: <class 'list'>
internal -> [] :: <class 'bool'>
external -> [] :: <class 'numpy.ndarray'>

疑问解答 #

1. 为何numpy空数组与空列表、空元组等空可迭代对象比较时返回空数组而非布尔值？ #

numpy的数组比较默认是元素级广播比较。当用==比较numpy数组和可迭代对象时，numpy会尝试将可迭代对象广播为与数组同形状的结构，再逐个元素比较。对于形状为(0,)的空数组np.empty(0)，空列表[]会被广播成同样形状的空数组，元素级比较后自然得到形状(0,)的空布尔数组。这是非空数组与非空可迭代对象元素级比较逻辑的自然延伸。

2. 为何在f-string内部执行该比较时，结果类型与外部执行不同？ #

这是因为f-string在格式化表达式结果时会做隐式布尔求值。numpy规定：只有非空且所有元素均为True的布尔数组，在布尔上下文中才会返回True，空布尔数组直接返回False。所以在f-string内部，np.empty(0) == []的结果会被隐式转换为bool类型，但它的字符串表示仍为[]，导致输出内容和外部的空数组看起来一致，但类型已经改变。你可以通过bool(np.empty(0) == [])验证，结果确实为False。

3. 为何numpy空数组与空字符串比较时返回False而非空数组？ #

空字符串是标量而非可迭代对象。numpy处理数组与标量的比较时，通常会将标量广播到数组每个元素位置做比较，但空数组没有元素可供比较，因此numpy退化为整体比较，返回标量布尔值False。新版本numpy会触发FutureWarning，提示未来可能会将此行为统一为返回空布尔数组，和可迭代对象的比较逻辑对齐。

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