Python 尝试简化csv中的读取,并将数组初始化为numpy数组

Python 尝试简化csv中的读取,并将数组初始化为numpy数组,python,arrays,csv,numpy,Python,Arrays,Csv,Numpy,我不熟悉使用Numpy。我正在尝试简化在特性中的读取和初始化numpy数组的设置 我希望功能名称包含列0-4的所有功能,_X包含所有行和列0-4,_y包含列5的所有行 我的代码可以工作,但它不像我希望的那样简洁易懂 import csv import numpy as np # read in the data as rows with open('data.csv', 'rb') as csvfile: _reader = csv.reader( csvfile, delimi

我不熟悉使用Numpy。我正在尝试简化在特性中的读取和初始化numpy数组的设置

我希望功能名称包含列0-4的所有功能,_X包含所有行和列0-4,_y包含列5的所有行

我的代码可以工作,但它不像我希望的那样简洁易懂

import csv 
import numpy as np

# read in the data as rows 
with open('data.csv', 'rb') as csvfile: 
    _reader = csv.reader( csvfile, delimiter =',',quotechar ='"') 

    # Read in the feature names into an array
    feature_names = _reader.next() 

    # Read the in the sample data
    _X, _y = [], []
    for row in _reader: 
        _X.append( row ) #read in plant  
        _y.append( row[ 5]) 

feature_names = np.array(feature_names) 
_X      = np.array( _X) 
_y      = np.array( _y)

_X = _X[:, [0,1,2,3,4]] 
_names = feature_names[[ 0,1,2,3,4]]

我真的很感谢你的帮助,并希望改善我的编码!提前感谢

在我看来,您正在实施该软件包。明确地 , 和
pandas.DataFrame
。看


你也可以使用
numpy.genfromtxt()

谢谢cyborg,我需要把东西放到numpy数组中作为随机森林的参数,我从来没有使用过熊猫,也可能没有它。我很感激任何进一步的建议,我尝试过使用它,但它不再是浮点数组了……(12.0,0.0,0.0,1.0,1.0),(18.0,1.0,0.0,1.0,0.0,0.0),(31.194181,0.0,0.0,0.0,0.0)],dtype=[('u刺的数量','