我正在尝试合并具有相同数量参数的两个数组。
输入:
first = [[650001.88, 300442.2, 18.73, 0.575, 650002.094, 300441.668, 18.775],
[650001.96, 300443.4, 18.7, 0.65, 650002.571, 300443.182, 18.745],
[650002.95, 300442.54, 18.82, 0.473, 650003.056, 300442.085, 18.745]]
second = [[1],
[2],
[3]]
我的预期输出:
final = [[650001.88, 300442.2, 18.73, 0.575, 650002.094, 300441.668, 18.775, 1],
[650001.96, 300443.4, 18.7, 0.65, 650002.571, 300443.182, 18.745, 2],
[650002.95, 300442.54, 18.82, 0.473, 650003.056, 300442.085, 18.745, 3]]
为此,我创建了简单的循环:
for i in first:
for j in second:
final += np.append(j, i)
我得到了我补充说我错过了一些东西。首先我的循环我非常慢。其次,我的数据有超过 200 万行要循环。所以我试图找到更快的方法,例如使用以下代码:
final = [np.append(i, second[0]) for i in first]
它的工作速度比前一个循环快得多,但它只附加第二个数组的第一个值。
你能帮助我吗?
最佳答案
使用 np.array
然后 np.concatenate
,
import numpy as np
first = np.array([[650001.88, 300442.2, 18.73, 0.575,
650002.094, 300441.668, 18.775],
[650001.96, 300443.4, 18.7, 0.65,
650002.571, 300443.182, 18.745],
[650002.95, 300442.54, 18.82, 0.473,
650003.056, 300442.085, 18.745]])
second = np.array([[1],
[2],
[3]])
np.concatenate((first, second), axis=1)
哪里
axis=1
意味着我们要水平连接。这对我行得通
关于arrays - 合并两个 numpy 数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43561622/