我正在尝试找到一个巧妙的小技巧,用于从二维数组中切割行/列并获得 (col_size x 1)
或 (1 x row_size)
.
有没有比在每次切片后使用 numpy.reshape()
更简单的方法?
干杯, 斯蒂芬
最佳答案
您可以在一次操作中切片并插入新轴。例如,这是一个二维数组:
>>> a = np.arange(1, 7).reshape(2, 3)
>>> a
array([[1, 2, 3],
[4, 5, 6]])
要切出单个 列(返回形状为 (2, 1)
的数组),以 None
作为第三维进行切片:
>>> a[:, 1, None]
array([[2],
[5]])
要切出单个 行(返回形状为 (1, 3)
的数组),以 None
作为第二个维度进行切片:
>>> a[0, None, :]
array([[1, 2, 3]])
关于python - Numpy - 从数组中切片二维行或列向量,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32764200/