python将csv数据文件转为灰度图像

收到老板给的好多csv数据文件让我转为灰度图像，带着很多的疑惑（我是做java的，对python有点兴趣，但是自己可以用来研究的时间并不是很多）研究出来的，可以根据自己的需求进行修改代码，代码实现了从本地的某个文件夹中获取所有后缀为csv文件的数据并转换成为名称和csv文件名称一致的jpg图片。
python小白记录一下自己的成长记录，不喜欢勿喷，如有不足，欢迎大佬提出宝贵的意见。

import csv
import numpy as np
import pandas as pd
from PIL import Image
import matplotlib.pyplot as plt
from scipy import misc
import cv2
import os

def get_files(path='D:\\matrixData', rule=".csv"):
    for fpathe, dirs, fs in os.walk(path):   # os.walk获取所有的目录
        for f in fs:
            filename = os.path.join(fpathe, f)
            if filename.endswith(rule):  # 判断是否是".csv"结尾
                with open(filename, encoding='UTF-8-sig') as csvfile:  #打开文件
                    list = np.loadtxt(csvfile, delimiter=",", skiprows=0)  #获取文件内的数据
                    img = Image.fromarray(np.uint8(list))
                    img.save(filename.replace(".csv", ".jpg"))

get_files()