主页 / 问题 / 648127
Accepted
Алексей Воронов
Asked:2020-04-04 15:42:42 +0000 UTC2020-04-04 15:42:42 +0000 UTC

添加两个不同维度的数组

  • 772

如何正确添加不同维度的数组？
有两个numpy数组。数组Wjout的维度为 [100, 100]，数组DW的维度为 [100:1]。DW数组被添加到Wjout数组的一列中。

Wjout[:,n] = Wjout[:,n] + DW

执行时，解释器给出以下错误：

Wjout[:,n] = Wjout[:,n] + DW
ValueError: 无法将输入数组从形状 (100,100) 广播到形状 (100)

python

1 个回答

  1. Best Answer

    试试这样：

    Wjout[:,n] += DW.ravel()

    例子：

    In [92]: a = np.arange(9).reshape(3,3)

In [93]: b = np.random.randint(10, size=3)[:, None]

In [94]: a
Out[94]:
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

In [95]: b
Out[95]:
array([[7],
       [7],
       [2]])

In [96]: a[:, 2] += b.ravel()

In [97]: a
Out[97]:
array([[ 0,  1,  9],
       [ 3,  4, 12],
       [ 6,  7, 10]])
    • 1

相关问题