python - 如何比较 Python 3 中的 numpy 字符串

Question

下面演示了问题:

import io
import numpy as np

a = np.loadtxt(io.StringIO("val1 val2\nval3 val4"), \
               dtype=np.dtype([("col1", "S10"), ("col2", "S10")]))
print("looks weired: %s"%(a["col1"][0]))
assert(a["col1"][0] == "val1")

我不明白应该如何比较字符串。在我的系统(numpy 1.6.2、python 3.2.2)上，输出如下所示:

>>> 
looks weired: b'val1'
Traceback (most recent call last):
  File "D:/..../bug_sample.py", line 7, in <module>
    assert(a["col1"][0] == "val1")
AssertionError

Answer 1

这不是numpy相关的:

>>> b"asd" == "asd"
False

在 Python 3 中，bytes 对象不等于 string。所以要么:

• 比较 b"val1" 而不是 "val1" 以便类型匹配，
• 将 bytes 对象解码为字符串(如 .decode('utf-8') 并与 "val1" 进行比较。