[np.newaxis, :]
时间: 2023-10-23 17:12:58 浏览: 98
探索 NumPy 的维度变换:np.newaxis 与 reshape 的奥秘
[np.newaxis, :] 是一个在 NumPy 中使用的索引技巧,用于在数组中添加一个新的轴。它的作用是将原来的数组从一维变为二维,并将原来的数据保持不变。
具体来说,对于一个形状为 (n,) 的一维数组,使用 np.newaxis 可以将其转换为形状为 (1, n) 的二维数组。在这个新的数组中,原来的数据按行排列。
举个例子,假设有一个一维数组 a,形状为 (4,),内容为 [1, 2, 3, 4]。使用 np.newaxis 可以将其转换为一个形状为 (1, 4) 的二维数组,内容仍然为 [1, 2, 3, 4]。
代码示例如下:
```python
import numpy as np
a = np.array([1, 2, 3, 4])
b = a[np.newaxis, :]
print(a.shape) # 输出 (4,)
print(b.shape) # 输出 (1, 4)
print(b) # 输出 [[1 2 3 4]]
```
这种索引技巧在进行一些需要多维数组的操作时非常有用,比如矩阵运算、广播等。希望这个解答能够帮到你!如果你还有其他问题,请继续提问。
阅读全文