Объединение двух одномерных массивов чисел
У меня есть два простых одномерных массивов в включает в себя. Я должен быть в состоянии объединить их с помощью и NumPy.конкатенат. Но я получаю эту ошибку для кода ниже:
TypeError: только массивы length-1 могут быть преобразованы в скаляры Python
код
import numpy
a = numpy.array([1, 2, 3])
b = numpy.array([5, 6])
numpy.concatenate(a, b)
почему?
4 ответов
строка должна быть:
numpy.concatenate([a,b])
массивы, которые вы хотите объединить, должны передаваться как последовательность, а не как отдельные аргументы.
numpy.concatenate((a1, a2, ...), axis=0)
соедините последовательность массивов вместе.
он пытался интерпретировать ваш b
как параметр оси, поэтому он жаловался, что не может преобразовать его в скаляр.
первый параметр concatenate
должно быть последовательность массивов объединить:
numpy.concatenate((a,b)) # Note the extra parentheses.
существует несколько возможностей для объединения массивов 1D, например,
numpy.r_[a, a],
numpy.stack([a, a]).reshape(-1),
numpy.hstack([a, a]),
numpy.concatenate([a, a])
все эти параметры одинаково быстры для больших массивов; для маленьких, concatenate
имеет небольшое преимущество:
сюжет был создан с perfplot:
import numpy
import perfplot
perfplot.show(
setup=lambda n: numpy.random.rand(n),
kernels=[
lambda a: numpy.r_[a, a],
lambda a: numpy.stack([a, a]).reshape(-1),
lambda a: numpy.hstack([a, a]),
lambda a: numpy.concatenate([a, a])
],
labels=['r_', 'stack+reshape', 'hstack', 'concatenate'],
n_range=[2**k for k in range(19)],
xlabel='len(a)',
logx=True,
logy=True,
)
альтернативный ist для использования короткой формы "concatenate", которая является либо " r_ [...]" или "когда[...] ", как показано в примере кода ниже (см. http://wiki.scipy.org/NumPy_for_Matlab_Users для получения дополнительной информации):
%pylab
vector_a = r_[0.:10.] #short form of "arange"
vector_b = array([1,1,1,1])
vector_c = r_[vector_a,vector_b]
print vector_a
print vector_b
print vector_c, '\n\n'
a = ones((3,4))*4
print a, '\n'
c = array([1,1,1])
b = c_[a,c]
print b, '\n\n'
a = ones((4,3))*4
print a, '\n'
c = array([[1,1,1]])
b = r_[a,c]
print b
print type(vector_b)
что приводит к:
[ 0. 1. 2. 3. 4. 5. 6. 7. 8. 9.]
[1 1 1 1]
[ 0. 1. 2. 3. 4. 5. 6. 7. 8. 9. 1. 1. 1. 1.]
[[ 4. 4. 4. 4.]
[ 4. 4. 4. 4.]
[ 4. 4. 4. 4.]]
[[ 4. 4. 4. 4. 1.]
[ 4. 4. 4. 4. 1.]
[ 4. 4. 4. 4. 1.]]
[[ 4. 4. 4.]
[ 4. 4. 4.]
[ 4. 4. 4.]
[ 4. 4. 4.]]
[[ 4. 4. 4.]
[ 4. 4. 4.]
[ 4. 4. 4.]
[ 4. 4. 4.]
[ 1. 1. 1.]]