.csv文件转tensor

原创已于 2023-03-13 15:45:03 修改 · 552 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#numpy #python #深度学习

于 2023-03-13 15:37:52 首次发布

该代码段定义了一个函数`csv_to_tensors`，它遍历指定文件夹中的所有CSV文件，使用pandas读取数据，然后转换为numpy数组，进一步转化为torch张量。最终，所有CSV文件的数据以张量列表的形式返回。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

直接给出代码：

import torch
import numpy as np
import os
import pandas as pd

def csv_to_tensors(folder_path):
    # 读取指定文件夹中的所有csv文件并将数据存储为张量格式
    csv_tensors = []
    for file_name in os.listdir(folder_path):
        if file_name.endswith('.csv'):
            # 读取csv文件
            file_path = os.path.join(folder_path, file_name)
            data = pd.read_csv(file_path)

            # 读取存为numpy数组再转tensor
            temp_array = np.array(data)
            tensor = torch.tensor(temp_array)
            # 将数据转换为张量格式并添加到列表中
            csv_tensors.append(tensor)

    # 返回csv文件的张量列表
    return csv_tensors

以上