Python: как загрузить файл CSV и извлечь из него определенные столбцы, которые будут использоваться в качестве входных данных?

У меня есть файл CSV, где каждая строка - это пациент, а каждый столбец - это «функция». Я хочу использовать этот код множественной линейной регрессии. Вместо набора данных из бостонского примера я хочу загрузить свой CSV-файл и использовать столбцы с 1 по 79 в качестве матрицы характеристик «данных» (X) и столбец 80 в качестве «целевого» вектора ответа (y). Как я могу это сделать? Я новичок в Python, поэтому приветствую любые советы.

      import matplotlib.pyplot as plt
import numpy as np
from sklearn import datasets, linear_model, metrics

# load the boston dataset
boston = datasets.load_boston(return_X_y=False)

# defining feature matrix(X) and response vector(y)
X = boston.data
y = boston.target

# splitting X and y into training and testing sets
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.4,
                                                    random_state=1)

# create linear regression object
reg = linear_model.LinearRegression()

# train the model using the training sets
reg.fit(X_train, y_train)

# regression coefficients
print('Coefficients: ', reg.coef_)

# variance score: 1 means perfect prediction
print('Variance score: {}'.format(reg.score(X_test, y_test)))

# plot for residual error

## setting plot style
plt.style.use('fivethirtyeight')

## plotting residual errors in training data
plt.scatter(reg.predict(X_train), reg.predict(X_train) - y_train,
            color = "green", s = 10, label = 'Train data')

## plotting residual errors in test data
plt.scatter(reg.predict(X_test), reg.predict(X_test) - y_test,
            color = "blue", s = 10, label = 'Test data')

## plotting line for zero residual error
plt.hlines(y = 0, xmin = 0, xmax = 50, linewidth = 2)

## plotting legend
plt.legend(loc = 'upper right')

## plot title
plt.title("Residual errors")

## method call for showing the plot
plt.show()

0 ответов

Другие вопросы по тегам