From the above dataset, I want to classify the column "Vehicle" into heavy vehicles and light vehicles using a Python loop and mapping function.
Vehicle
Color
car
gray
bus
blue
van
white
bicycle
yellow
bike
green
tractor
red
car
blue
bus
gray
car
black
bike
orange
bus
blue
car
red
tractor
blue
bike
yellow
bus
gray
tractor
green
bike
blue
tractor
white
bus
red
bike
gray
I have a large dataset and I am looking for something that will split the vehicle column into two columns: Heavy Vehicles and Light Vehicles.
Please solve ASAP.
In [1]: import numpy as np
import pandas as pd
In [2]: Nfrom pandas import ExcelFile
df = pd.read_excel("Example.xlsx")
In [3]: +PK
Out[3]:
vehicle Color
0 car KeB
1 sns blue
2 van white
3 bicycle yellow
4 bike green
5 tractor red
6 car blue
7 bus gray
8 car black
9 bike orange
10 bus blue
11 car red
12 tractor blue
13 bike yellow
14 bus Keub
15 tractor green
16 bike blue
17 tractor white
18 bus red
19 bike Kerb