从.txt文件读入整数到numpy数组

时间:2015-08-28 16:36:19

标签: python numpy text genfromtxt

如本问题所述,我想从简单的.txt数组中读取4个整数 genfromtxt : read ints from space separated .txt file

但我想把它当作2D 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数组?

1 个答案:

答案 0 :(得分:3)

只需使用:

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

您正在创建(int,int,int,int)的一维数组,但您真正想要的是np.int32的二维数组。

相关问题