我目前正在使用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％。
我可以编写一个函数，以便根据用户输入所需的细分以及与每个细分相关的百分比来动态创建细分吗？
谢谢！