Python：處理Robot Navigation數據集的標籤

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)