You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
7 lines
287 B
7 lines
287 B
4 years ago
|
|
||
|
def split_data_frame(data_frame, column):
|
||
|
""" splits a data frame on unique column values """
|
||
|
values = data_frame[column].unique()
|
||
|
masks = {value: (data_frame[column] == value) for value in values}
|
||
|
return {value: data_frame[mask] for value, mask in masks.items()}
|