【NumPy】1つの配列と同じサイズで全ての要素が「0」「1」の配列を生成

この記事では、Python言語とNumPyを用いて、1つの配列と同じサイズで全ての要素が「0」や「1」の配列を生成する方法をソースコード付きで解説します。

全ての要素の値が同じな配列

NumPy配列では、特定の配列と同じサイズで全ての要素が「0」や「1」の配列を生成する機能が用意されています。

書式

説明
X = numpy.zeros_like(A) 配列Aと同じサイズで全ての要素が0の配列Xを生成します。
X = numpy.ones_like(A) 配列Aと同じサイズで全ての要素が1の配列Xを生成します。

ソースコード

サンプルプログラムのソースコードです。

# -*- coding: utf-8 -*-
import numpy as np

def main():
    # 配列の宣言・初期化
    X = np.array([[1, 2, 3],[4, 5, 6]])
    
    # 結果表示
    print(np.zeros_like(X))
    print(np.ones_like(X))


if __name__ == '__main__':
    main()
[[0 0 0]
 [0 0 0]]

[[1 1 1]
 [1 1 1]]

おすすめ記事

Python入門 サンプル集
NumPy入門 サンプル集

シェア&フォローお願いします!