質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

88.62%

配列の大きさをそろえる方法?

解決済

回答 1

投稿

  • 評価
  • クリップ 0
  • VIEW 3,396
退会済みユーザー

退会済みユーザー

配列の大きさをそろえる方法がわからないです。
以下のコードを実行すると

# coding: utf-8

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import os

import numpy as np


a = np.arange(5)
b = np.arange(24).reshape((4,3,2))


while a.ndim > b.ndim:
    b = b[np.newaxis,:]
    print(b)
    print(b.size)
    print(b.ndim)
while b.ndim > a.ndim:
    a = a[np.newaxis,:]
    print(a)
    print(a.size)
    print(a.ndim)
[[0 1 2 3 4]]
5
2
[[[0 1 2 3 4]]]
5
3


と出力結果が出ました。

a = np.arange(5)
b = np.arange(24).reshape((4,3,2))


の出力結果は

[0 1 2 3 4]
[[[ 0  1]
  [ 2  3]
  [ 4  5]]

 [[ 6  7]
  [ 8  9]
  [10 11]]

 [[12 13]
  [14 15]
  [16 17]]

 [[18 19]
  [20 21]
  [22 23]]]


で、while文でa・b2つの変数の中で使われている
np.newaxis の役割がわかりません。
np.newaxisは配列の次元を増やす役割をするので
特に

 b = b[np.newaxis,:]


でbの次元数が減っていることが理解できません。
ここではaとbの次元数をそろえることをおこなっているとおもうのですが、
どうしてnp.newaxisにより次元数がそろえられるのでしょうか?

  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 1

checkベストアンサー

+1

bの次元は減っていません。
最初のループでは、bの次元>aの次元なので処理されません。
次のループでは、aの次元がbの次元数まで増やされます。

import numpy as np

def print_log(name,a):
    print("%s size[%s],dim[%s],shape[%s]---"%(name,a.size,a.ndim,a.shape))
    print(a)
    print("---")

a = np.arange(5)
b = np.arange(24).reshape((4,3,2))
print_log("a",a)
print_log("b",b)

print("b -> a")
while a.ndim > b.ndim:
    b = b[np.newaxis,:]
    print_log("b",b)

print("a -> b")
while b.ndim > a.ndim:
    a = a[np.newaxis,:]
    print_log("a",a)

print_log("a",a)
print_log("b",b)

結果

a size[5],dim[1],shape[(5,)]---
[0 1 2 3 4]
---
b size[24],dim[3],shape[(4, 3, 2)]---
[[[ 0  1]
  [ 2  3]
  [ 4  5]]

 [[ 6  7]
  [ 8  9]
  [10 11]]

 [[12 13]
  [14 15]
  [16 17]]

 [[18 19]
  [20 21]
  [22 23]]]
---
b -> a
a -> b
a size[5],dim[2],shape[(1, 5)]---
[[0 1 2 3 4]]
---
a size[5],dim[3],shape[(1, 1, 5)]---
[[[0 1 2 3 4]]]
---
a size[5],dim[3],shape[(1, 1, 5)]---
[[[0 1 2 3 4]]]
---
b size[24],dim[3],shape[(4, 3, 2)]---
[[[ 0  1]
  [ 2  3]
  [ 4  5]]

 [[ 6  7]
  [ 8  9]
  [10 11]]

 [[12 13]
  [14 15]
  [16 17]]

 [[18 19]
  [20 21]
  [22 23]]]
---

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 88.62%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

同じタグがついた質問を見る