在 ndarray python 中获取第一个维度

Question

假设我有一个像这样的 ndarray ：

    a = [[20 43 61 41][92 23 43 33]]

我想获取这个 ndarray 的第一个维度。所以我尝试这样的事情：

    a[0,:]

我希望它会返回这样的结果：

    [[20 43 61 41]]

但我收到了这个错误：

   TypeError: 'numpy.int32' object is not iterable

任何人都可以帮我解决这个问题吗？

Answer 1

使用切片：

>>> import numpy as np
>>> a = np.array([[20, 43, 61, 41], [92, 23, 43, 33]])
>>> a[:1]  # OR a[0:1]
array([[20, 43, 61, 41]])
>>> print(a[:1])
[[20 43 61 41]]