【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

# 配列の宣言・初期化
X = np.array([[1, 2, 3],[4, 5, 6]])

# 結果表示
print(np.zeros_like(X))
"""
[[0 0 0]
 [0 0 0]]
"""

print(np.ones_like(X))

"""
[[1 1 1]
 [1 1 1]]
"""
関連記事
1 Python入門 サンプル集
2 NumPy入門 サンプル集

コメント