将元素添加到numpy 1D数组中

时间:2015-06-29 02:55:29

标签: python numpy

我有一个numpy数组(a)。如何在其中添加两个元素?

import numpy as np
a = np.array([5,6,7])

b = 3
c = 4

result = np.hstack((b,c,a))
print result

使用@BHAT IRSHAD的答案

我必须这样做才能与np.savetxt一起使用;我想把结果写成一行。但不能。

np.savetxt('fo.txt', result, delimiter = ',', fmt = '%s')

1 个答案:

答案 0 :(得分:1)

np.hstack只接受一个参数而你传递的是3个,所以result = np.hstack((b,c,a))会起作用。

演示:

>>> a = np.array([5,6,7])
>>> b = 3
>>> c = 4
>>> np.hstack((b,c,a))
array([3, 4, 5, 6, 7])