Python Numpy：从数组中提取一行

Question

我正在尝试使用从 Numpy 数组中提取一行

t = T[153,:]

但我发现其中的大小T是 (17576, 31)，大小t是 (31,) - 尺寸不匹配！

我需要t有尺寸 (,31) 或 (1,31) 以便我可以将其输入到我的函数中。我试过转置，但没有用。

谁能帮我解决一个简单的问题？

非常感谢

Answer 1

您可以使用切片符号提取行：

t = T[153:154,:]    # will extract row 153 as a 2d array

示例：

T = np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12]])

T[1,:]
# array([5, 6, 7, 8])

T[1,:].shape
# (4,)

T[1:2,:]
# array([[5, 6, 7, 8]])

T[1:2,:].shape
# (1, 4)