В настоящее время я сегментирую фрейм данных, используя numpy.split. У меня вопрос, могу ли я переписать свой код, чтобы динамически получать пользовательский ввод из функции, чтобы определить количество сегментов и процентное соотношение сегментов.
Мой текущий код:
import pandas as pd
import numpy as np
df_1, df_2, df_3 = np.split(df, [int(0.5*len(df)), int(0.75*len(df))])
Этот код хорошо работает для создания трех фреймов данных с первыми 50% строк в первом фрейме данных и 25% в каждом из двух других фреймов данных.
Могу ли я написать функцию для динамического создания сегментов, основываясь на вводе пользователем того, сколько сегментов им нужно, и процентах, связанных с каждым сегментом?
Спасибо!