使用索引的数组内的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.]])格式，即；就像我们在[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.]])