如何用Pandas DataFram读取包含像素值的csv文件

2024-10-02 12:23:12 发布

您现在位置:Python中文网/ 问答频道 /正文

我的csv文件如下所示:

im1000100101     0
im1011100101     1

第一列是图像的像素值,第二列是该图像的类。如何使用pd.read_csv()将每个像素保存在单独的列中。我想我的数据帧是这样的:

^{pr2}$

Tags: 文件csv数据图像read像素pdpr2
1条回答
网友
1楼 · 发布于 2024-10-02 12:23:12

使用^{}

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

相关问题 更多 >

    热门问题