python - 如何使用 Pandas DataFrame 读取包含图像像素值的 csv 文件

Question

我的 csv 文件如下所示:

im1000100101     0
im1011100101     1

第一列是图像的像素值，第二列是图像的类别。如何使用 pd.read_csv() 将每个像素保存在单独的列中。我希望我的 DataFrame 看起来像这样:

px-1  px-2  px-3  px-4  px-5 px-6  px-7  px-8  px-9  px-10  label
1     0     0     0     1    0     0     1     0     1      0 
1     0     1     1     1    0     0     1     0     1      1

Answer 1

使用read_fwf :

import pandas as pd
from pandas.compat import StringIO

temp=u"""im1000100101     0
im1011100101     1"""
#after testing replace 'StringIO(temp)' to 'filename.csv'
N = 12
df = pd.read_fwf(StringIO(temp), header=None, widths=[1] * N + [6], usecols=range(2,13))
df.columns = ['px-{}'.format(x+1) for x in df.columns[:-1]] + ['label']
print (df)
   px-1  px-2  px-3  px-4  px-5  px-6  px-7  px-8  px-9  px-10  label
0     1     0     0     0     1     0     0     1     0      1      0
1     1     0     1     1     1     0     0     1     0      1      1