def get_trace_attributes(log):
"""
Gets the attributes at the trace level of a log object
Parameters
----------------
log
Log object
Returns
---------------
trace_attributes_list
List of attributes at the trace level
"""
from pm4py.util import constants
if check_is_dataframe(log):
check_dataframe_columns(log)
return [x for x in list(log.columns) if x.startswith(constants.CASE_ATTRIBUTE_PREFIX)]
else:
from pm4py.statistics.attributes.log import get
return list(get.get_all_trace_attributes_from_log(log))
def get_trace_attributes(log: Union[EventLog, pd.DataFrame]) -> List[str]:
"""
Gets the attributes at the trace level of a log object
Parameters
----------------
log
Log object
Returns
---------------
trace_attributes_list
List of attributes at the trace level
"""
if type(log) not in [pd.DataFrame, EventLog, EventStream]: raise Exception("the method can be applied only to a traditional event log!")
from pm4py.util import constants
if check_is_pandas_dataframe(log):
check_pandas_dataframe_columns(log)
return [x for x in list(log.columns) if x.startswith(constants.CASE_ATTRIBUTE_PREFIX)]
else:
from pm4py.statistics.attributes.log import get
return list(get.get_all_trace_attributes_from_log(log))
