「NumPy」配列ndarrayの要素をスカラー値とと比較するサンプル

2022年7月4日

環境
PyCharm 2021.3
Python 3.9.7

構文
変数名 = np.arange(配列の数).reshape((行数, 列数))
ndarrayをスカラー値と比較すると、bool値（True, False）を要素としてもつndarrayが返されます。
<や==, !=などで要素を比較します。

使用例

# coding: utf-8

import numpy as np

cft = np.arange(6).reshape((2, 3))
print("配列の要素")
print(cft)

print("6以下の要素の比較値bool")
print(cft < 6)

print("==で要素を比較する結果")
print(cft % 2 == 0)

実行結果

配列の要素
[[0 1 2]
 [3 4 5]]
6以下の要素の比較値bool
[[ True  True  True]
 [ True  True  True]]
==で要素を比較する結果
[[ True False  True]
 [False  True False]]