在numpy.ndarray的数组中numpy.ndarray的list和int居然可以直接进行比较并把结果作为索引

numpy.ndarray的数组中numpy.ndarray的list和int居然可以直接进行比较并把结果作为索引

import numpy as np
b = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
a = np.array([0, 0, 0, 1, 1, 1, 2, 2, 2])
print(type(a))
for i in range(3):
    # 真神奇了，这个居然也能比较
    # numpy.ndarray的list和int类型的在数组中
    # 居然可以直接比较并当做索引使用
    print(b[a == i])

for i in range(3):
    print(a)
    print('\n')
    print(a == i)
    # 说明ndarray和int进行比较的时候会首先将i
    # 复制与a等长的n分

输出结果：

<class 'numpy.ndarray'>
[1 2 3]
[4 5 6]
[7 8 9]

[0 0 0 1 1 1 2 2 2]
# 这里应该对应的添加i为：
# [0 0 0 0 0 0 0 0 0]
# 所以对比输出的结果是下边这个
[ True  True  True False False False False False False]
# 这个同理需要添加下边这一行
# [1 1 1 1 1 1 1 1 1]
[0 0 0 1 1 1 2 2 2]
# 所以对比的结果就是下边这一行
[False False False  True  True  True False False False]
# 同理
# [2 2 2 2 2 2 2 2 2]
[0 0 0 1 1 1 2 2 2]
# 对比结果
[False False False False False False  True  True  True]