python - 使用 pandas 将描述符保存到 csv

Question

我的代码有问题。我想将 HOG 描述符保存到 csv 文件中。

代码:

import cv2
import pandas as pd

def getHogFromImage(im):
   hog = cv2.HOGDescriptor()
   h = hog.compute(cv2.imread(im))
   return h

h1 = getHogFromImage("apple.jpg")
h2 = getHogFromImage("peach.jpg")

apple = pd.DataFrame("1," + h1) #any idea how do it?
peach = pd.DataFrame("0," + h2)

apple.to_csv('foo.csv', index=False, header=None, sep=",", line_terminator="\n")
peach.to_csv('foo.csv', index=False, header=None, sep=",", line_terminator="\n")

我的 foo.csv 文件

0.15200253
0.16865349
0.0909358
0.13702433
0.15322956
0.14372425
0.18107016
0.12029551
0.065096505
0.20796815
0.28659236
0.15984383
0.12754892
0.15886153
.
.
.

但我需要以这种格式将描述符保存到文件中

1, 0.15200253, 0.16865349, 0.0909358, 0.13702433 .....
0, 0.16865349, 0.0909358, 0.13702433

所以所有描述符数据都将在一行中，用“，”分隔。但是当图像为苹果时，第一个值为“1”，当图像为桃色时，第一个值为 0。

Answer 1

假设您的 hog 值是浮点数列表，您可以通过将 h 值放入 [] 中来在数据框构造函数中创建列表列表

df = pd.DataFrame([h1,h2]).to_csv('foo.csv', index=False, header=None, sep=",", line_terminator="\n")