rapids/src/models/workflow_example/merge_features_and_targets_...

11 lines
459 B
Python

import pandas as pd
import numpy as np
index_columns = ["local_segment", "local_segment_label", "local_segment_start_datetime", "local_segment_end_datetime"]
sensor_features = pd.read_csv(snakemake.input["cleaned_sensor_features"], index_col=index_columns)
targets = pd.read_csv(snakemake.input["targets"], index_col=index_columns)
data = pd.concat([sensor_features, targets[["target"]]], axis=1, join="inner")
data.to_csv(snakemake.output[0], index=True)