web-dev-qa-db-ja.com

NumPy配列のサイズ変更と拡張

私はPythonで作業していて、次のような NumPy 配列を持っています:

[1,5,9]
[2,7,3]
[8,4,6]

次のように伸ばすにはどうすればよいですか?

[1,1,5,5,9,9]
[1,1,5,5,9,9]
[2,2,7,7,3,3]
[2,2,7,7,3,3]
[8,8,4,4,6,6]
[8,8,4,4,6,6]

これらはほんの一部の配列の例です。実際には、これらだけでなく、いくつかのサイズの配列のサイズを変更します。

私はこれが初めてで、私がしなければならないことに頭を包むことができないようです。

28
Matthew

@KennyTMの答えは非常に巧妙で、実際にあなたのケースで機能しますが、配列を拡張するためのもう少し柔軟性を提供する可能性のある代替手段として、np.repeatを試してください。

>>> a = np.array([[1, 5, 9],
              [2, 7, 3],
              [8, 4, 6]])

>>> np.repeat(a,2, axis=1)
array([[1, 1, 5, 5, 9, 9],
       [2, 2, 7, 7, 3, 3],
       [8, 8, 4, 4, 6, 6]])

したがって、これにより、1つの軸に沿って繰り返し、複数の軸に沿って(必要に応じて)、np.repeat呼び出しをネストするだけで済みます。

>>> np.repeat(np.repeat(a,2, axis=0), 2, axis=1)
array([[1, 1, 5, 5, 9, 9],
       [1, 1, 5, 5, 9, 9],
       [2, 2, 7, 7, 3, 3],
       [2, 2, 7, 7, 3, 3],
       [8, 8, 4, 4, 6, 6],
       [8, 8, 4, 4, 6, 6]])

最初の行または列の繰り返し回数を変更することもできます。たとえば、最後の行以外の各行を2回繰り返す場合は、次のようにします。

>>> np.repeat(a, [2,2,1], axis=0)
array([[1, 5, 9],
       [1, 5, 9],
       [2, 7, 3],
       [2, 7, 3],
       [8, 4, 6]])

ここで、2番目の引数がlistの場合、行ごとに(axis=0であるため、この場合は行)が各行に対して繰り返されることを指定します。

27
dtlussier
>>> a = numpy.array([[1,5,9],[2,7,3],[8,4,6]])
>>> numpy.kron(a, [[1,1],[1,1]])
array([[1, 1, 5, 5, 9, 9],
       [1, 1, 5, 5, 9, 9],
       [2, 2, 7, 7, 3, 3],
       [2, 2, 7, 7, 3, 3],
       [8, 8, 4, 4, 6, 6],
       [8, 8, 4, 4, 6, 6]])
14
kennytm

残念ながら、numpyは分数ステップを許可していません(私が知る限り)。これが回避策です。 Kennyのソリューションほど賢くはありませんが、従来のインデックス作成を利用しています。

>>> a = numpy.array([[1,5,9],[2,7,3],[8,4,6]])
>>> step = .5
>>> xstop, ystop = a.shape
>>> x = numpy.arange(0,xstop,step).astype(int)
>>> y = numpy.arange(0,ystop,step).astype(int)
>>> mg = numpy.meshgrid(x,y)
>>> b = a[mg].T
>>> b
array([[1, 1, 5, 5, 9, 9],
       [1, 1, 5, 5, 9, 9],
       [2, 2, 7, 7, 3, 3],
       [2, 2, 7, 7, 3, 3],
       [8, 8, 4, 4, 6, 6],
       [8, 8, 4, 4, 6, 6]])

(dtlussierのソリューションの方が優れています)

1
Paul