python - 如何将多个 .npy ndarray 文件连接到一个数据帧中？

Question

我有一个包含 24 个“.npy”文件的文件夹，其中包含 ndarrays（人脸编码的 ndarray）。我想创建一个遍历它们的 for 循环并将其附加到一个新的数据帧中。我怎么做？我试图这样做，但它会引发错误：无法连接零维数组

def convert_dataframe(data):
    return pd.DataFrame(np.concatenate(data))

embed_path = "<path to folder>"
dataframe = pd.DataFrame()
for embeddings in embed_path:
    embed = convert_dataframe(embeddings)
    embed.append(dataframe)

任何的意见都将会有帮助。

标签： pythonpandasdataframe

Answer 1

尝试以下方法：

import os
import numpy as np

embed_path = './path_to_folder/'       # Use forward slash (/) at the end

arr = []
for file in os.listdir(embed_path):
    temp = np.load(embed_path + file)
    arr.append(temp)

df = pd.DataFrame(a)
print(df)