首页 > 解决方案 > 如何将结果从 np.mean 转换为 csv?

问题描述

所以我写了一个脚本来获取文件夹中每个图像的平均灰度值。当我执行 print(np.mean(img) 时,我得到了终端上的所有值。但我不知道如何将值获取到 csv 数据。

import glob
import cv2
import numpy as np
import csv
import pandas as pd


files = glob.glob("/media/rene/Windows8_OS/PROMON/Recorded Sequences/6gParticles/650rpm/*.png")
for file in files:
    img = cv2.imread(file)
    finalArray = np.mean(img)
    print(finalArray)

到目前为止它有效,但我需要在 csv 数据中有值。我尝试了 csvwriter 和 pandas 但没有设法获取包含灰度值的文件。

标签: pandascsvimage-processing

解决方案


这是你要找的吗?

files = glob.glob("/media/rene/Windows8_OS/PROMON/Recorded Sequences/6gParticles/650rpm/*.png")
mean_lst = []
for file in files:
    img = cv2.imread(file)
    mean_lst.append(np.mean(img))

pd.DataFrame({"mean": mean_lst}).to_csv("path/to/file.csv", index=False)

推荐阅读