Python：处理Robot Navigation数据集的标签

最新推荐文章于 2025-03-13 14:42:17 发布

DeniuHe

最新推荐文章于 2025-03-13 14:42:17 发布

阅读量415

点赞数

分类专栏： Python学习

本文链接：https://blog.csdn.net/DeniuHe/article/details/103639318

版权

Python学习专栏收录该内容

239 篇文章

订阅专栏

本文介绍了一种将机器人导航原始数据进行预处理的方法，通过读取CSV文件中的传感器读数，将其转换为适用于机器学习模型的数据格式。具体包括将分类标签转化为数值型标签，并将处理后的数据保存为新的CSV文件。

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

import numpy as np
import pandas as pd

data = np.array(pd.read_csv(r'E:\dataset\未处理数据集\Robot Navigation\sensor_readings_24.csv',header=None))
X = data[:,:-1]
y = data[:,-1]
label = []
n = len(y)

for i in range(n):
    if y[i] == 'Slight-Right-Turn':
        label.append(0)
    elif y[i] == 'Move-Forward':
        label.append(1)
    elif y[i] == 'Slight-Left-Turn':
        label.append(2)
    elif y[i] == 'Sharp-Right-Turn':
        label.append(3)

label = np.vstack(label)
Data = np.hstack((X,label))
Data = pd.DataFrame(Data)

Data.to_csv(r"E:\dataset\未处理数据集\Robot Navigation\Robot_Navigation_24.csv",header=None,index=None)