python - 切片索引在 numpy 数组中如何工作

Question

假设我们有一个数组

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

现在我有以下内容

row_r1 = a[1, :]
row_r2 = a[1:2, :]

print(row_r1.shape)
print(row_r2.shape)

我不明白为什么 row_r1.shape 是 (4,) 而 row_r2.shape 是 (1,4)

它们的形状不应该都等于(4,)吗？

Answer 1

我喜欢这样想。第一种方式 row[1, :] 表示获取第 1 行上的所有值，如下所示:

返回:数组([5,6,7,8])

形状

(4,) numpy 数组中的四个值。

作为第二个row[1:2, :]，状态是获取索引 1 和索引 2 之间的数据切片:

返回:

array([[5, 6, 7, 8]]) 注意:双括号

形状

(1,4) np.array 中一行中有四个值。