Qual é o equivalente de “zip ()” no numpy do Python?
Eu estou tentando fazer o seguinte, mas com matrizes numpy:
x = [(0.1, 1.), (0.1, 2.), (0.1, 3.), (0.1, 4.), (0.1, 5.)]
normal_result = zip(*x)
Isso deve dar um resultado de:
normal_result = [(0.1, 0.1, 0.1, 0.1, 0.1), (1., 2., 3., 4., 5.)]
Mas se o vetor de entrada é um array numpy:
y = np.array(x)
numpy_result = zip(*y)
print type(numpy_result)
Ele (esperado) retorna um:
<type 'list'>
A questão é que vou precisar transformar o resultado de volta em um array numpy depois disso.
O que eu gostaria de saber é o que acontece se houver uma função numpy eficiente que evite essas transformações de ida e volta?