Question

我有两个numpy数组：

A = np.array([1, 3, 5, 7])
B = np.array([2, 4, 6, 8])

我想从两者结合得到以下内容：

C = [1, 2, 3, 4, 5, 6, 7, 8]

我可以使用zip来获得一些东西，但不是我正在寻找的东西：

>>> zip(A, B)
[(1, 2), (3, 4), (5, 6), (7, 8)]

如何明智地组合两个numpy数组？

我对每个阵列中的50,000个元素进行了快速测试（100,000个组合元素）。结果如下：

User Ma3x:      Time of execution: 0.0343832323429      Valid Array?:  True
User mishik:    Time of execution: 0.0439064509613      Valid Array?:  True
User Jaime:     Time of execution: 0.02767023558        Valid Array?:  True

^{使用Python 2.7，Windows 7企业版64位，英特尔酷睿i7 2720QM @ 2.2 Ghz Sandy Bridge测试，8 GB内存}

Answer 1

使用np.insert：

>>> A = np.array([1, 3, 5, 7])
>>> B = np.array([2, 4, 6, 8])
>>> np.insert(B, np.arange(len(A)), A)
array([1, 2, 3, 4, 5, 6, 7, 8])

Answer 2

您也可以使用切片：

C = np.empty((A.shape[0]*2), dtype=A.dtype)
C[0::2] = A
C[1::2] = B

Answer 3

>>> import numpy as np
>>> A=np.array([1,3,5,7])
>>> B=np.array([2,4,6,8])
>>> C=np.dstack([A,B])
>>> D=C.reshape((1,8))[0]
>>> D
array([1, 2, 3, 4, 5, 6, 7, 8])

Answer 4

有些答案建议排序，但由于你想要将它们组合在一起，因此元素排序不会达到相同的结果。

这是一种方法

C = []
for elem in zip(A, B):
    C.extend(elem)

Answer 5

你可以试试这个：

C = sorted(A.tolist() + B.tolist())

A.tolist()将产生[1, 3, 5, 7]
B.tolist()将产生[2, 4, 6, 8]
A.tolist() + B.tolist() - [1, 3, 5, 7, 2, 4, 6, 8]
sorted(...) - [1, 2, 3, 4, 5, 6, 7, 8]

没有排序：

C = [y for x in zip(A, B) for y in x]

如何在python中组合两个numpy数组元素？

5 个答案: