我正在阅读CSV格式:
import numpy as np
features = np.genfromtxt('train.csv',delimiter=',',usecols=(1,2))
输出数据为:
[[ - 1。 -1。] [1. -1。] [-1。 1.] [1. -1。]]。请参阅
之后的点1
和-1
train.csv
0,-1,-1
-1,1,-1
0,-1,1
1,1,-1
-1,1,-1
0,1,-1
答案 0 :(得分:1)
正如评论中所述:np.genfromtxt
默认情况下只是将您的数据转换为浮点数(see the default dtype argument in the function signature)。如果要强制输出为整数,只需在genfromtxt中指定dtype=np.int
:
features = np.genfromtxt('train.csv',delimiter=',',usecols=(1,2),dtype=np.int)