У меня есть CSV-файл и хочу извлечь столбец и использовать INT как массив

Question

У меня есть CSV-файл. Одно из названий столбцов - mpg. Я читаю файл таким образом.

data = pd.read_csv('auto-mpg.csv')   
var_1 = data['mpg']  
train_y = numpy.asarray(var_1)  

Как мне скопировать данные данных var1 в массив train_y?

Answer 1

Новое: ОК. Пропустил, что это тренировочный массив, вероятно, это тот же вопрос Как создать тестовые и обучающие образцы из одного кадра данных с пандами?

Старый: Некоторые примеры данных могут помочь, но вы можете использовать панды для копирования и форматирования данных при первом проходе. это то, что я думаю, вы ожидаете, что это будет работать как.

import pandas as pd
import numpy

    df = pd.read_csv('auto-mpg.csv',converters={"mpg": lambda x: numpy.asarray(x)})
    train_y = df.mpg