从.txt文件中读取int到numpy数组中

Question

我试图从这个问题中描述的简单。txt数组中读取4个整数Genfromtxt:从。txt文件中读取int

但是我希望它是二维numpy数组

def read_data():
    data = np.genfromtxt('Skin_NonSkin.txt', dtype=(int, int, int, int))
    print type(data)
    print data.shape
    print data[0]
    print type(data[0])
    print data[0].shape
    print data[0][1]
    return data

它给我

<type 'numpy.ndarray'>
(245057L,)
(74, 85, 123, 1)
<type 'numpy.void'>
()
85

那么如何正确读取数据或将其转换为形状为(245057,4)的2D nampy数组?

Answer 1

直接使用

data = np.genfromtxt('Skin_NonSkin.txt', dtype=np.int32)

您正在创建(int,int,int,int)的1D数组，但您真正想要的是np.int32的2D数组。