CSV文件转换为灰度图像

1.  
   from PIL import Image
2.  
   import pandas as pd
3.  
   import numpy as np
4.  
    
5.  
    
6.  
   data_path = r'E:\硕士\东南大学齿轮箱数据\gearset\Health_30_2.csv'
7.  
   data = pd.read_csv(data_path, delimiter='\t')
8.  
   data = data.values
9.  
    
10.  
    image_array = data[:1024]
11.  
    array1 = []
12.  
    for i, _ in enumerate(image_array):
13.  
    # print(image_array[i])
14.  
    array1.append(image_array[i][:8])
15.  
     
16.  
    print(type(array1), type(image_array))
17.  
    array1 = np.array(array1)
18.  
    array1 *= 255.0 # 变换为0-255的灰度值
19.  
    im = Image.fromarray(array1)
20.  
    im = im.convert('L') # 这样才能转为灰度图，如果是彩色图则改L为‘RGB’
21.  
    im = im.resize(size=(32,32))
22.  
    im.show()

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

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

这篇好文章是转载于：学新通技术网