如何在 numpy.matrix 中添加 numpy.array？

Question

我有一個矩陣，通過這種方式定義：

pop = (np.random.randint(3, size=(qtd_individuos, qtd_equipamentos))) + 1

我想在此矩陣中添加一個新數組。

vetor2 = np.ones((qtd_equipamentos))*2

我已經以這種方式嘗試了一些事情，使用 np.append、np.vstack、np.stack、np.concatenate、np.add...沒有任何效果...使用 append 將其列入列表太容易了。我不確定我是否正在嘗試做對。 我發現的示例與此略有不同。

np.insert(pop, vetor1)
np.stack((pop, vetor2))

Answer 1

我是這樣解決的：

pop = np.append(pop, [vetor1], axis = 0) # The square brackets are important!

Answer 2

你也可以用vstack做到這一點：

np.vstack([pop, vetor2])

建議確保類型相同，因此您可以使用例如：

vetor2 = np.ones(qtd_equipamentos, dtype=int)*2

您的 output 具有 dtype dtype('int32') ，這與沒有此修復的dtype不同。