在一个循环中加入numpy

Question

我无法将数据与计算值WLD1连接起来。所以我的数据是200x3。在添加计算的WLD1并将其作为列添加到数据中之后，我需要一个200x4维的数据集。我得到了WLD1的一列值，但无法将其与数据连接起来。我得到维度误差

请帮帮我

拥有200X3的数据集。请在下面找到代码

import numpy as np
import pandas as pd
N = 200
mean = [180,200,400]
cov = [[0.3, 0.2, 0.1],[0.2, 0.2, 0.1], [0.1, 0.1, 0.4]]
data = np.random.multivariate_normal(mean, cov, N)
for i in data:
    wLD1 = np.dot(np.dot(np.transpose(i - mean), np.linalg.inv(cov)), (i - mean))

Answer 1

好吧，您的变量wLD1是一个浮点而不是数组，因此，您不能将它连接到您的数据

你能做的是：

mean = np.array(mean)
cov = np.array(cov)
wLD1 = np.dot(np.dot(np.transpose(data - mean), np.linalg.inv(cov)), (data - mean))

然后，您将wLD1作为shape200的数组。您将能够将它连接到您的数据