python numpyで繰り返された配列を作成する

python numpyで繰り返された配列を作成する

pythonで、ライブラリnumpyのrepeatを使用して、繰り返された配列を作成するサンプルコードを記述してます。pythonのバージョンは3.8.5を使用してます。

環境

  • OS windows10 pro 64bit
  • python 3.8.5

numpyインストール

numpyをインストールされていない方は、pipでインストールしておきます。

pip install numpy

# Successfully installed numpy-1.19.4

repeat使い方

repeatを使用すると、繰り返された配列を作成するが可能です。

import numpy as np

arr = np.array(["a", "b", "c"])

print(arr.repeat(3))

# ['a' 'a' 'a' 'a' 'b' 'b' 'b' 'b' 'c' 'c' 'c' 'c']

これを利用すると行が同じ値の二次元配列が簡単に作成できます。

import numpy as np

arr = np.array(["a", "b", "c"])

print(arr.repeat(3).reshape(3, 3))

# [['a' 'a' 'a']
#  ['b' 'b' 'b']
#  ['c' 'c' 'c']]