python - 将元素插入到 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.insert 中嵌套 np.append 很可能不是执行此操作的 pythonic 方法。

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])