can say this is a slam dunk
This commit is contained in:
@@ -21,6 +21,13 @@ def write_joined_df(left, right, lsuf="new_key"):
|
|||||||
return merged
|
return merged
|
||||||
|
|
||||||
|
|
||||||
def slam_dunk(dataset, column, size):
|
def slam_dunk(dataset, column, size, labels):
|
||||||
dataset["dunked"] = pd.qcut(dataset[column], q=3)
|
min_value = dataset[column].min()
|
||||||
return dataset["dunked"]
|
max_value = dataset[column].max()
|
||||||
|
bins = np.linspace(min_value, max_value, size)
|
||||||
|
|
||||||
|
dunked_column = "bin_" + column
|
||||||
|
dataset[dunked_column] = pd.cut(
|
||||||
|
dataset[column], bins=bins, labels=labels, include_lowest=True
|
||||||
|
)
|
||||||
|
return dataset[dunked_column]
|
||||||
|
|||||||
Reference in New Issue
Block a user