Numpy数组有头尾方法吗？

Question

我将一个csv文件加载到“数据集”中，并尝试执行dataset.head（），但它报告了一个错误。如何检查numpy数组的头或尾？没有指定特定的行？

Answer 1

对于类似头的函数，您可以使用切片数组dataset[:10]。

对于类似尾巴的函数，您可以使用切片数组dataset[-10:]。

警告：dataset[-1:] 返回数组的最后一个元素，但 dataset[-0:] 返回整个数组，因此 tail(0) 返回所有内容而不是不返回任何内容。请参阅下面我的答案，它解决了这个问题。 (2认同)

Answer 2

您可以对任何可迭代的 Python 执行此操作。

Python 3.x ( https://www.python.org/dev/peps/pep-3132/ ) 中的PEP-3132可以使用该*符号表示可迭代对象的“其余部分”。

做你想做的事：

>>> import numpy as np
>>> np.array((1,2,3))
array([1, 2, 3])
>>> head, *tail = np.array((1,2,3))
>>> head
1
>>> tail
[2, 3]