reverse segment.py

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import glob
import os
mile_to_meter = 1609.34
Input_data_path = "Crew Districts/"
csv_files = glob.glob(os.path.join(Input_data_path, '*.csv'))

for csv_name in csv_files:
    print("Loading: " + csv_name)
    input_file_name = csv_name
    output_file_name = "Reverse " + csv_name
    original = pd.read_csv(input_file_name)
    print(original)
    reverse_grade = np.zeros(len(original))
    reverse_curve_direction = np.zeros(len(original))
    MP_BEG = np.zeros(len(original))
    MP_END = np.zeros(len(original))
    reverse_segment = original[["LINE_SEGMENT","BEG MP","END MP","SPEED","GRADE_TO_NEXT","DIRECTION","DEG OF CURVE","LENGTH (M)","ELEV (M)"]]

    for i in range(len(original)):
        reverse_segment.loc[i,"END MP"] = original.loc[i,"BEG MP"]
        reverse_segment.loc[i,"BEG MP"] = original.loc[i,"END MP"]
        reverse_segment.loc[i,"GRADE_TO_NEXT"] = original.loc[i,"GRADE_TO_NEXT"]*(-1)
        if original.loc[i,"DIRECTION"] == "R":
            reverse_segment.loc[i,"DIRECTION"] = "L"
        if original.loc[i,"DIRECTION"] == "L":
            reverse_segment.loc[i,"DIRECTION"] = "R"
    reverse_segment = reverse_segment[::-1].reset_index(drop=True)
    print(input_file_name + " Done")
    reverse_segment.to_csv(output_file_name)