将CSV文件转换为灰度图像

from PIL import Image
import pandas as pd
import numpy as np


data_path = r'E:\硕士\东南大学齿轮箱数据\gearset\Health_30_2.csv'
data = pd.read_csv(data_path, delimiter='\t')
data = data.values

image_array = data[:1024]
array1 = []
for i, _ in enumerate(image_array):
    # print(image_array[i])
    array1.append(image_array[i][:8])

print(type(array1), type(image_array))
array1 = np.array(array1)
array1 *= 255.0  # 变换为0-255的灰度值
im = Image.fromarray(array1)
im = im.convert('L')  # 这样才能转为灰度图，如果是彩色图则改L为‘RGB’
im = im.resize(size=(32,32))
im.show()

 

 

上图是数据集的具体格式，数据与数据之前的空格是制表符 \t 

 这是我打印image_array数据的内容，数据前八个是有具体值的，后面部分显示nan，至于为什么显示nan我目前还不知道