将元素插入numpy数组的开头和结尾

Question

我有一个numpy数组:

import numpy as np
a = np.array([2, 56, 4, 8, 564])

我想添加两个元素:一个在数组的开头88,一个在结尾,77.

我可以这样做:

a = np.insert(np.append(a, [77]), 0, 88)

所以a最终看起来像:

array([ 88,   2,  56,   4,   8, 564,  77])

问题是:这样做的正确方法是什么？我觉得自己像一个嵌套np.append在一个np.insert很可能不是Python的方式来做到这一点.

Answer 1

另一种方法是使用numpy.concatenate.示例 -

np.concatenate([[88],a,[77]])

演示 -

In [62]: a = np.array([2, 56, 4, 8, 564])

In [64]: np.concatenate([[88],a,[77]])
Out[64]: array([ 88,   2,  56,   4,   8, 564,  77])

Answer 2

您可以使用np.concatenate-

np.concatenate(([88],a,[77]))

Answer 3

关于什么：

a = np.hstack([88, a, 77])

Answer 4

您可以将索引列表传递给np.insert：

>>> np.insert(a,[0,5],[88,77])
array([ 88,   2,  56,   4,   8, 564,  77])

或者，如果您不知道数组的长度，则可以array.size用来指定数组的结尾：

>>> np.insert(a,[0,a.size],[88,77])
array([ 88,   2,  56,   4,   8, 564,  77])