我有一个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

新：好，错过了一个训练阵列，这可能是相同的问题， How do I create test and train samples from one dataframe with pandas?

旧： 一些示例数据会有所帮助，但是您可以使用熊猫在第一次通过时对数据进行翻录和格式化。这是我想像的，您希望它能像这样工作。

import pandas as pd
import numpy

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