Python Numpy : Extracting a row from an array

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)