使用索引在数组内部的numpy ndarray

Question

我有一个numpy.ndarray - a

和a[0]给了我：

array([0., 0., 0., ..., 0., 0., 0.])

a[0:1]也给出了：

array([[0., 0., 0., ..., 0., 0., 0.]])

我如何使a[0]格式为array([[0., 0., 0., ..., 0., 0., 0.]]) ie;在数组内部的数组格式，就像我们做[list[0]]

Answer 1

如果你想重塑数组，那么第一个轴（每一行）的每个元素都是一个形状（1，m）的数组，而不仅仅是（m，），你可以做以下事情：

a = a[:,np.newaxis,:]

这样，（n，m）数组转换为形状（n，1，m），并使用a[0]返回第一行：

array([[0., 0., 0., ..., 0., 0., 0.]])