numpy 數組的尺寸為 1 和沒有該尺寸有什么區別

Question

我可能有點困惑。 但我想知道 x[2,3] 和 y[2,3,1] 之間有什么區別（相同的數組，但具有大小為 1 的額外維度）。

它們是相同的還是它們之間存在差異。

Answer 1

讓我們舉一個二維的例子

# shape (2,)
a = np.array([0,1])
# shape (2,1)
b = np.array([[3],[4]])

您可以將a視為具有 2 列的單行（實際上是 1D 向量），將b數組視為具有 1 列的 2 行。

讓我們嘗試添加它們：

a+a
# addition on a single dimension
# array([0, 2])

b+b
# also common dimensions
# array([[6],
#        [8]])

a+b
# different dimensions with one of common size
# addition will be broadcasted to generate a (2,2) shape
# array([[3, 5],
#        [4, 6]])