f70e71d5c7fdc8e25391e54e74c3402fb323ad5c,examples/plot_employee_salaries.py,,,#,45

Before Change

// the rest will have a standard encoding
data_path = fetching.get_data_dir()
fetching.fetch_employee_salaries()
data_file = os.path.join(data_path, "employee_salaries", "rows.csv")
df = pd.read_csv(data_file).astype(str)
df["Current Annual Salary"] = [float(s[1:]) for s
                               in df["Current Annual Salary"]]
df["Year First Hired"] = [int(s.split("/")[-1])
                          for s in df["Date First Hired"]]

target_column = "Current Annual Salary"
y = df[target_column].values.ravel()

After Change

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// and carry out some basic preprocessing:
df["Current Annual Salary"] = df["Current Annual Salary"].str.strip("$").astype(
    float)
df["Date First Hired"] = pd.to_datetime(df["Date First Hired"])
df["Year First Hired"] = df["Date First Hired"].apply(lambda x: x.year)

target_column = "Current Annual Salary"
y = df[target_column].values.ravel()

Instances