python:numpy 數組切片一維數組

Question

我將一個數組初始化為 a=numpy.array([1,2,3])。 在運行語句 print(a[0,:]) 時，它顯示一個錯誤。 這種切片方法是否僅適用於 2d arrays？

Answer 1

你可以解決這個問題

a = a[np.newaxis, :]

在打印之前，使其成為1 x 3陣列而不是具有形狀 (3,)。 顯然，這只有在您還需要其他多維 arrays 的打印語句並希望使其以通用方式工作時才有意義。

Answer 2

只需將“a[0,:]”替換為“a[0:]”即可。

import numpy as np

a = np.array([1, 2, 3])
print(a[0:])