Проблема возврата ошибки Numpy при добавлении массивов

Question

«ValueError: все измерения входного массива, кроме оси конкатенации, должны точно совпадать» - это ошибка, возникающая при попытке добавить значения.PFB код. x - это набор данных размером [16754,3], а a - это массив только одного с размером [16754,1]. Насколько я понимаю, ось точно совпадает.

# Importing the libraries
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

# Importing the dataset
dataset = pd.read_csv('data_monthly_rainfall.csv')
x = dataset.iloc[:, [0,1,2]].values
y = dataset.iloc[:, 3].values

# Apending a coloumn y with 1 for the equation
import statsmodels.api as sm
a = np.ones((16754, 0)).astype(int)
x = np.append(arr = a,values = x, axis = 1)

Может кто-нибудь сказать, пожалуйста, что я делаю здесь не так? Я очень новичок в Python и ML, на этапе обучения. Пожалуйста, дайте мне знать, если вам нужна дополнительная информация.

Ссылка на набор данных

Answer 1

Проблема в том, что форма x имеет вид (16755, 3), а вы создаете a с формой (16754, 1). Несоответствие происходит в строках. Измените размеры a на (16755, 1) :

a = np.ones((16755, 1)).astype(int)

Этого можно избежать, сохранив количество строк в переменной.

m = x.shape[0]
a = np.ones((m, 1)).astype(int)
x = np.append(arr = a,values = x, axis = 1)