web-dev-qa-db-ja.com

2つのnumpy配列をマージする

同じ数の引数を持つ2つの配列をマージしようとしています。

入力：

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)

私は何かが足りないということを埋めました。まず第一に、私のループは非常に遅いです。第二に、私のデータはかなり2ミリ行以上ループしました。そこで、たとえば次のコードを使用して、より高速な方法を見つけようとしました。

final = [np.append(i, second[0]) for i in first]

前のループよりもはるかに高速に動作しますが、2番目の配列の最初の値のみを追加します。手伝って頂けますか？

arrayspython-3.xnumpymerge
6
Losbaltica

使用する 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は、水平方向に連結することを意味します。

それは私のために働く

12
mforez

使用する np.column_stack：

import numpy as np

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]]

np.column_stack([first, second])

リストとして必要な場合は、tolistメソッドを使用します。

np.column_stack([first, second]).tolist()
6
user9389968