Присоединяясь к NumPy в петле

Question

Я не могу объединить данные с вычисленными значениями WLD1. Таким образом, мои данные составляют 200 X 3. После добавления вычисленного WLD1 и добавления его в виде столбца в данных мне нужен набор данных измерения 200 x 4. Я получаю столбец значений для 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 в виде массива формы 200. Вы сможете объединить его с вашими данными.